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

https://www.acmicpc.net/problem/1316

1316번: 그룹 단어 체커

 

문제

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

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

입력

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

출력

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

 

---

내 풀이

# 31120kb 44ms

n = int(input())
ans = 0

for i in range(n):
  word = list(input())
  flag = True
  checked_list = []

  for check in word:
    cnt = 0
    if check not in checked_list:
      if word.count(check) == 1:
        pass
      else:
        for idx in range(len(word)-1):
          if word[idx] == check and word[idx+1] == check:
            cnt +=1
        if cnt == (word.count(check) -1):

          pass
        else:
          flag = False
      checked_list.append(check)

  if flag:
    ans+=1

print(ans)

 

메모

스터디원의 좋은 코드가 있어서 참고로 메모하였다. 

# 31120KB, 48ms
N = int(input())
count = 0
for _ in range(N):
    word = input()
    if list(word) == sorted(word, key=word.find):
        count += 1
print(count)

sorted(word, key=word.find)를 사용하면 알파벳 종류에 따라 정렬해주는데

'happyp' 라는 단어가 있다면 'happpy'로 등장하는 알파벳 종류에 따라 정렬해준다.

두 단어가 일치하지 않으면 위의 'happyp'의 'p'의 경우처럼 알파벳이 연속 되지 않는 것이 존재하니 조건을 충족하지 못해 count되지 않는다.