[Python]1316번, 문자열에서 특정 패턴 개수 세기

문제

 

그룹 단어란 단어에 존재하는 모든 문자에 대해서, 각 문자가 연속해서 나타나는 경우만을 말한다. 

예를 들면, ccazzzzbb는 c, a, z, b가 모두 연속해서 나타나고, 

kin도 k, i, n이 연속해서 나타나기 때문에 그룹 단어이지만, aabbbccb는 b가 떨어져서 나타나기 때문에 그룹 단어가 아니다.

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

 

입력

 

첫째 줄에 단어의 개수 N이 들어온다. 

N은 100보다 작거나 같은 자연수이다. 

둘째 줄부터 N개의 줄에 단어가 들어온다. 

단어는 알파벳 소문자로만 되어있고 중복되지 않으며, 길이는 최대 100이다.

 

출력

 

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

 

풀이

 

제목에 적은 것처럼 주어진 문자열에서 원하는 패턴이 존재하는지 아닌지를 이용해 문제를 풀었다.

 

순서는 이렇다.

 

• 주어진 문자열의 개수 n에서 시작
• i번째 문자열 str 순회 시작
• str의 j번째 문자열의 개수 구하기(str.count(j))
• 이어서 j에 그 개수만큼 곱한 문자열(j * str.count(j))이  str에 존재하는지 확인(in str)
• 만약 없다면 그룹 단어가 아니기 때문에 n에서 1을 빼고 조건문 탈출해서 다음 문자열로 이동

말로 쓰니까 코드보다 길어지는데.. 코드는 아래와 같다.

import sys

n = int(sys.stdin.readline())

for i in range(n):
    str = sys.stdin.readline()
    for j in str:
        if j * str.count(j) not in str:
            n -= 1
            break

print(n)

문자열에서 원하는 단어의 개수나 포함여부를 체크할 수 있다는 것이 굉장히 편리하게 느껴졌다.

 

문자열 단원도 끝!