2024.7.20 백준(Python) 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

예제 입력 3 

5
ab
aa
aca
ba
bb

예제 출력 3 

4

예제 입력 4 

2
yzyzy
zyzyz

예제 출력 4 

0

 

 

정답 코드1

result = 0  # 그룹 단어의 개수를 저장할 변수

N = int(input())  # 입력받을 단어의 수

for _ in range(N):  # N번 반복
    check = 1  # 그룹 단어 여부를 체크하는 변수, 1은 그룹 단어임을 의미
    word = input()  # 단어를 입력받음
    word_list = []  # 문자들을 저장할 리스트

    # 입력받은 단어의 각 문자를 순회
    for i in range(len(word)):
        if not word_list:  # word_list가 비어있으면
            word_list.append(word[i])  # 현재 문자를 추가
        elif word_list[len(word_list) - 1] != word[i]:  # 이전 문자와 현재 문자가 다르면
            word_list.append(word[i])  # 현재 문자를 추가

    # word_list의 각 문자를 순회
    for i in range(len(word_list)):
        num = word_list.count(word_list[i])  # 현재 문자의 개수를 셈
        if num != 1:  # 문자가 한 번 이상 나타나면
            check = 0  # 그룹 단어가 아님을 표시
            break  # 더 이상 체크할 필요 없으므로 루프 종료
    
    if check == 1:  # 그룹 단어이면
        result += 1  # 결과값에 1을 더함

print(result)  # 그룹 단어의 총 개수를 출력

 

정답 코드2(개선)

# 입력받을 단어의 수
N = int(input())
# 그룹 단어의 개수를 저장할 변수, 초기값은 N으로 설정
result = N

# N번 반복
for _ in range(N):
    # 단어를 입력받음
    word = input()
    # 단어의 길이 - 1만큼 반복
    for i in range(len(word) - 1):
        # 현재 문자와 다음 문자가 같으면
        if word[i] == word[i + 1]:
            continue  # 다음 반복으로 넘어감
        # 현재 문자가 다음 문자 이후에 또 존재하면
        elif word[i] in word[i + 1:]:
            result -= 1  # 그룹 단어 개수를 줄임
            break  # 더 이상 체크할 필요 없으므로 루프 종료

# 최종 그룹 단어의 개수를 출력
print(result)

 

• 불필요한 리스트 생성 및 중복 검사 루프를 피한다.
• count 메서드를 사용하지 않고, 단순히 in 연산과 비교만으로 그룹 단어를 판단한다.
• 루프 구조가 더 간단하고, 불필요한 작업을 최소화하여 실행 시간이 짧다.