문제
그룹 단어란 단어에 존재하는 모든 문자에 대해서, 각 문자가 연속해서 나타나는 경우만을 말한다. 예를 들면, ccazzzzbb는 c, a, z, b가 모두 연속해서 나타나고, kin도 k, i, n이 연속해서 나타나기 때문에 그룹 단어이지만, aabbbccb는 b가 떨어져서 나타나기 때문에 그룹 단어가 아니다.
단어 N개를 입력으로 받아 그룹 단어의 개수를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 단어의 개수 N이 들어온다. N은 100보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 단어가 들어온다. 단어는 알파벳 소문자로만 되어있고 중복되지 않으며, 길이는 최대 100이다.
출력
첫째 줄에 그룹 단어의 개수를 출력한다.
| 예제 입력 | 예제 출력 |
|---|---|
| 3 happy new year | 3 |
| 4 aba baba bacbac a | 1 |
| 9 aaa aaazbz babb aazz azbz aabbaa abacc aba zzaz | 1 |
풀이
설계
어떤 단어를 입력으로 받았을 때, 해당 단어의 문자들이 '연속적'인지를 체크하는 것이 가장 주요한 요소라고 판단하였다.
1. 문자열을 받아 가장 첫 번째 단어를 group 리스트에 넣는다.
2. 문자열의 2번째 문자부터 하나씩 꺼내 group 리스트의 '마지막' 요소와 비교한다.
2-1. 두 문자가 동일할 경우 group 리스트에 추가한다.
-> 연속된 문자
2-2. 두 문자가 같지는 않으나 group 리스트에 존재하지 않는 경우 역시 group 리스트에 추가한다.
-> 새로운 문자
2-3. 두 문자가 같지 않고 group 리스트에 이미 존재하는 경우 반복문을 빠져나간다.
-> 연속적이지도 않고 이미 앞에 등장했던 문자
3. 입력받은 단어와 group 리스트의 길이를 비교해 동일하면 cnt의 값을 1 증가시킨다.
코드
cnt = 0
n = int(input())
for _ in range(n):
word = input()
group = [word[0]]
for c in word[1:]:
if c == group[-1]:
group.append(c)
elif (c != group[-1]) and (c not in group):
group.append(c)
else:
break
if len(word) == len(group):
cnt += 1
print(cnt)참고 풀이
훨씬 더 간결하면서도 파이썬의 함수 중 특기할 만한 사항이 몇 가지 있는 코드를 발견하여 정리한다.
cnt = 0
n = int(input())
for _ in range(n):
word = input()
cnt += list(word) == sorted(word, key=word.find)
print(cnt)특기 사항
- int 자료형에 True 혹은 False를 더하여 값을 증가시킬 수도 있다.
count = 0
count += True
print(count) # 1이를 이용하여 비교 조건에 따라 값을 증가시키고자 할 때 if문을 사용하는 것보다 더 간결하게 표현할 수 있다.
count = 0
word = "abcde"
count += "a" in word
print(count) # 1- 일반적으로 sorted는 내림차순 정렬이 기본으로 설정되어 있지만 key 파라미터에 인자를 전달하면 해당 인자를 기준으로 정렬한다. 이때 str.find를 인자로 전달하면 해당 문자열에 등장한 단어의 순서를 그대로 유지하면서 요소를 정렬하는데, 상기 풀이 코드에서는 이를 이용하여 입력된 단어의 '그룹 단어' 여부를 체크하였다. 실제 코드를 보면 이해하기 쉽다.
word = "addcbd"
print(f"word : {list(word)}")
print(f"sorted(word) : {sorted(word)}")
print(f"sorted(word, key=word.find) : {sorted(word, key=word.find)}")
# word : ['a', 'd', 'd', 'c', 'b', 'd']
# sorted(word) : ['a', 'b', 'c', 'd', 'd', 'd']
# sorted(word, key=word.find) : ['a', 'd', 'd', 'd', 'c', 'b']