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

문제

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

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

입력

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

출력

첫째 줄에 그룹 단어의 개수를 출력한다.

(틀린)풀이

import sys
input = sys.stdin.readline

cnt = 0
words = []
N = int(input())
for _ in range(N):
    words.append(input().rstrip())

for word in words:
    for i in range(len(word)):
        # a의 첫번째 인덱스 + (a의 갯수 - 1) 에 해당하는 인덱스도 a여야지만 불연속적인 a가 없는 것
        # ex. 'aabbbccb'에서 b의 첫번째 인덱스 2 + b의 갯수 4 - 1 = 5번 인덱스도 b여야지 b가 다 붙어있는 것
        if word[word.find(word[i]) + word.count(word[i]) - 1] != word[i]:
            # 그룹단어가 아닌 것 발견하면 1 추가
            cnt += 1
            break

# 전체 단어에서 그룹단어 아닌 것 제외하고 출력
print(N - cnt)

아주 옛날에 한 번 풀었던 문제를 오랜만에 다시 풀었는데, 틀렸다.
예전에 풀었던 방법과는 다른 방식으로,
각 단어들word을 순회하면서
각 스펠링의 갯수를 카운트해
해당 단어에서 그 갯수만큼 스펠링이 연속적으로 붙어있는지를 체크했다.

문제에 딸려있는 테스트케이스도 모두 통과했고,
자체적으로 케이스를 만들어 돌려봤지만 모두 통과했는데..
왜 틀렸는지 모르겠다. 좀 더 살펴봐야겠다..

오답이유 발견

예외 : yyayyyz
마지막 yyy부분이 연속적이라 그룹 단어로 인식한다.

(맞는)풀이

import sys
input = sys.stdin.readline

N=int(input())
count=0
for _ in range(N):
    word=input().rstrip()
    error=0
    for t in range(len(word)-1):
        if word[t] != word[t+1]:
            new_word=word[t+1:]
            if new_word.count(word[t]) > 0:
                error=1
    if error == 0:
        count+=1
print(count)

각 단어에서 스펠링이 바뀌는 순간('a' -> 'b') 거기('b')부터 그 뒤까지 단어를 모두 잘라서 새로운 단어를 만들고
새로운 단어에서 비교대상이었던 스펠링('a)이 하나 이상 존재하는지 체크한다.
존재하면 그룹단어 X

Reference

https://www.acmicpc.net/problem/1316 - 백준 온라인 저지 Problem 1316