[백준] 1316번 그룹 단어 체커 파이썬 (python)

1316번: 그룹 단어 체커

 

문제

---

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

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

 

입력

---

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

 

 

 

정답코드

---

def isitg(word):
    seen = []  # 이미 본 문자를 추적
    prev_char = ''  # 이전 문자를 저장
    for char in word:
        if char in seen:  # 이전에 본 문자인 경우
            if char != prev_char:  # 이전 문자와 다르면 그룹 단어가 아님
                return 0
        else:
            seen.append(char)  # 새 문자를 추가
            prev_char = char  # 이전 문자를 업데이트
    return 1  # 모든 검사를 통과하면 그룹 단어

totalg = 0
cnt = int(input())

for i in range(cnt):
    word = input()
    totalg += isitg(word)

print(totalg)

 

풀이

---

생각보다 까다로운 문제이다. if문의 중첩을 어떻게 잘 이해하고 구현하느냐가 문제 해결의 관건이었다. 

입력받은 단어를 매개변수로 하는 함수 isitg는 다음과 같은 구조를 가진다:

 

1. 이미 본 문자를 추적하는 리스트 seen을 선언하고 초기화한다. 
2. 현재 인덱스 바로 이전 인덱스의 단어를 저장할  prev_char를 선언하고 초기화한다. 
   
   
3. word안의 각 글자를 순서대로 볼 건데, 
4. 만약 해당 글자가 seen 안에 존재한다면
   
   
   1. 그리고 만약 해당 글자가 이전 문자인 prev_char가 아니라면 
   2. 0을 반환한다. (그룹 단어가 아님)
5. 만약 해당 글자가 seen 안에 존재하지 않는다면
   
   
   1. 이미 본 문자의 리스트 안에 해당 글자를 append 한다. 
   2. append후 prev_char안에 해당 글자를 넣어준다. 
6. 단어가 모든 조건을 통과했다면 1을 반환한다.
7. 이제 나머지 코드를 구현해주면 된다.

 

느낀 점

---

내가 처음 작성했던 코드의 isitg함수는 다음과 같았다.

 

def isitg(word):
    lista = []

    for indexa, alpha in enumerate(word):
        if alpha in lista: 
            if indexa == 0:
                lista.append(alpha)
            elif indexa >= 1 and alpha == lista[indexa - 1]:
                continue
            else:
                return 0
        else:
            lista.append(alpha)  
    return 1

 

이 함수는 특정 조건이 만족되면 indexeorror를 일으킨다.  alpha == lista[indexa - 1]: 의 범위를 벗어난 리스트의 연산으로  9, aaa를 입력할 시 연속으로 반복되는 단어의 처리가 불가능하다. prev_char와 같이 이전 단어를 훨씬 더 간결하게 찾는 매커니즘을 생각하는 데 오래 걸려서 살짝 아쉬운 마음이 든다.