[코딩 테스트 연습(파이썬/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이다.

[출력]

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

 

[아이디어]

나온 적이 있는 단어가 연속이 아닌 상태에서 또 등장하면 0을 return하는 함수를 만든다.

 

n = int(input())
def g(s):
    d=dict()
    for i, x in enumerate((s:=' ' + s)):
        if x not in d:
            d[x] = 1
        elif (x in d) and s[i-1]==x: pass
        else: return 0
    return 1
answer = 0
for _ in range(n):
    answer += g(input())
print(answer)

이렇게 풀긴 했는데, 더 간단한 방법이 있었다. sorted()로 정렬된 리스트를 만들 때 key에 옵션을 줄 수 있는데, 던지는 인자가 str인 경우 그 str에 대한 find 함수도 인자로 던질 수 있나보다. 즉 해당 str에 포함된 문자들에 대해 find() 함수의 결과 값인 '앞에서 몇 번째에서 처음 해당 문자가 발견되는가'를 기준으로 정렬하기 때문에, 아래의 결과가 만들어진다.

즉 이를 이용하면, 다음과 같이 풀 수 있다.

n = int(input())
answer = 0
def g(s):
    return list(s)==sorted(s,key=s.find)
for _ in range(n):
    answer += g(input())
print(answer)