[알고리즘] 백준 1316 그룹 단어 체커

문제

그룹 단어란 단어에 존재하는 모든 문자에 대해서, 각 문자가 연속해서 나타나는 경우만을 말한다. 예를 들면, ccazzzzbb는 c, a, z, b가 모두 연속해서 나타나고, kin도 k, i, n이 연속해서 나타나기 때문에 그룹 단어이지만, aabbbccb는 b가 떨어져서 나타나기 때문에 그룹 단어가 아니다.
단어 N개를 입력으로 받아 그룹 단어의 개수를 출력하는 프로그램을 작성하시오.

  • 입력
    첫째 줄에 단어의 개수 N이 들어온다. N은 100보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 단어가 들어온다. 단어는 알파벳 소문자로만 되어있고 중복되지 않으며, 길이는 최대 100이다.

내 풀이

from itertools import groupby
def checkgroup(word):
    group = [key for key, item in groupby(word)]
    values =[(k, [i for i in range(len(word)) if word[i] == k]) for k in group]

    groupword = 0
    for items in values:
        if items[1].__len__() == 0:
            groupword += 1
            continue        
        for index in range(1, items[1].__len__()):
            if items[1][index]-items[1][index-1] >1:
                groupword = 0
                return groupword
        groupword += 1
    return groupword    
        
        
        
count = int(input())
words = [input() for _ in range(count)]
result = 0
for _ in words:
    groupcount = checkgroup(_)
    if groupcount != 0:
        result+=1
    
print(result)

복잡하게 구현된 것 같은 느낌도 살짝 든다. 더 효율적인 알고리즘이 있을거같은 느낌?? 그래도 고민고민하다가 마침내 푼 것이 기분이 좋다!! 각 문자별 인덱스 배열을 가지고 그 배열이 연속적인 수인지를 판별했다. 그 과정에서 리스트를 위한 groupby를 알게 되었다.

출처

[백준] 1316번: 그룹 단어 체커

좋은 웹페이지 즐겨찾기