[백준] 1316. 그룹 단어 체커 python

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

문제

그룹 단어란 단어에 존재하는 모든 문자에 대해서, 각 문자가 연속해서 나타나는 경우만을 말한다. 예를 들면, ccazzzzbb는 c, a, z, b가 모두 연속해서 나타나고, kin도 k, i, n이 연속해서 나타나기 때문에 그룹 단어이지만, aabbbccb는 b가 떨어져서 나타나기 때문에 그룹 단어가 아니다.

단어 N개를 입력으로 받아 그룹 단어의 개수를 출력하는 프로그램을 작성하시오.

풀이과정

alpha라는 딕셔너리에 나온 알파벳이면 key를 알파벳으로, value를 index로 저장한다. 이후 같은 알파벳이 등장한 경우에 alpha[key] 값과 현재 인덱스 차이가 1인 경우에만 진행하고, 아닌 경우(떨어져서 나타나는 경우)에는 중단하도록 하였다.

소스코드

import sys

input = sys.stdin

n = int(input.readline().strip())  # 단어의 개수
words = list(map(str, input.readline().strip()) for _ in range(n))
cnt = 0
for word in words:
    alpha = dict()
    flag = True
    for idx, w in enumerate(word):
        if w in alpha.keys():  # alpha에 존재하는 단어이면
            if abs(alpha[w] - idx) == 1:
                alpha[w] = idx
            else:
                flag = False
                break
        else:
            alpha[w] = idx
    if flag:  # 그룹 단어이면
        cnt += 1
print(cnt)