[백준 1316] 그룹 단어 체크(Python)

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

1316번: 그룹 단어 체커

문제

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

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

입력

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

출력

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

예제 입력 1

3 happy new year

예제 출력 1

3

예제 입력 2

4 aba abab abcabc a

예제 출력 2

1

풀이 설명

1. n만큼 단어를 입력받아 배열 arr에 저장한다.

2. arr을 탐색

3. arr의 각 원소 a를 탐색

  3-1. c가 이전 문자와 다르고, dic에 이미 있으면 그룹 단어가 아니다.

  3-2. a의 각 단어 c를 딕셔너리에 저장한다. (key는 c, value는 개수)

Python3 코드

n = int(input())
arr = []
count = 0

for i in range(n):
    s = input()
    arr.append(s)

for a in arr:
    dic = {}
    last = a[0]
    isbreak = False
    for c in a:
        if last != c and dic.get(c, -1) != -1: #이전 문자와 다르고, dic에 이미 있으면 그룹단어가 아님
            isbreak = True
            break
        dic[c] = dic.get(c, 0) + 1
        last = c
    if not isbreak:
        count += 1

print(count)