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

Jiyeahhh·2021년 11월 22일

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

문제

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

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

입력

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

출력

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

풀이

아이디어

연속되지 않은 곳에서 문자가 다시 나타날 경우 그 문자열은 그룹 단어가 아님
이것만 생각하고 다음과 같이 코드 작성

count = 0

for i in range(int(input())):
    word = input()
    for j in word:
        if j in word[word.index(j)+2:]:
            count -= 1
            break
    count += 1

print(count)

연속되지 않은 문자열 내에 해당 문자가 있을 경우 break
count -= 1을 한 이유는 break 실행해도 가장 바깥 for 문은 돌기 때문에 무조건 count += 1이 실행됨. 그래서 1 빼준 거임... ~~뭔가 굉장히 비효율적인 거 같은..~~

주어진 예제에서는 다 맞게 돌아갔는데 오답 처리됨

생각해보니까 만약 문자열이 aaab라면? 위 코드로는 a가 연속적으로 나타나 있어도 ab와 비교하기 때문에 if문 실행됨
그래서 그 사이에도 똑같은 문자가 아닐 조건을 추가함

count = 0

for i in range(int(input())):
    word = input()
    for j in word:
        if j in word[word.index(j)+2:]:
            # 추가!!!
            if word[word.index(j)+1] != j:
                count -= 1
                break
    count += 1

print(count)

하지만 오답

이제 aaaba 같은 경우가 문제
만약 연속된 문자가 다를 때 현재 문자 이후 문자열을 새롭게 정의
그 새로운 문자열 내에 현재 문자가 있을 경우 해당 문자열은 그룹 문자가 아님

코드

n = int(input())
count = 0

for i in range(n):
    word = input()
    for j in range(len(word)-1):
        if word[j] != word[j+1]:
            new = word[j+1:]
            if word[j] in new:
                count -= 1
                break
    count += 1

print(count)