[백준] 2941: 크로아티아 알파벳 (Python)

문제

예전에는 운영체제에서 크로아티아 알파벳을 입력할 수가 없었다. 따라서, 다음과 같이 크로아티아 알파벳을 변경해서 입력했다.

č	c=
ć	c-
dž	dz=
đ	d-
lj	lj
nj	nj
š	s=
ž	z=

예를 들어, ljes=njak은 크로아티아 알파벳 6개(lj, e, š, nj, a, k)로 이루어져 있다. 단어가 주어졌을 때, 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.

dž는 무조건 하나의 알파벳으로 쓰이고, d와 ž가 분리된 것으로 보지 않는다. lj와 nj도 마찬가지이다. 위 목록에 없는 알파벳은 한 글자씩 센다.

입력

첫째 줄에 최대 100글자의 단어가 주어진다. 알파벳 소문자와 '-', '='로만 이루어져 있다.

단어는 크로아티아 알파벳으로 이루어져 있다. 문제 설명의 표에 나와있는 알파벳은 변경된 형태로 입력된다.

출력

입력으로 주어진 단어가 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.

 

풀이

어떻게 처리해야 할지 고민을 하다가

1. 크로아티아 알파벳 리스트를 만들고

2. 크로아티아 알파벳 개수만을 세는 리스트를 생성하여 크로아티아 알파벳 개수 세기

- dz= 안에 z= 가 포함되어 있어 중복으로 세어지므로 중복 제거

3. 기존 문자열에서 크로아티아 알파벳 제거

4. 문자열 길이 + 크로아티아 알파벳 개수 리스트 요소 합 출력

 

으로 생각을 했고 채점이 중간정도 되다가 틀렸다. 아마 시간초과로 틀린 것 같다.

import sys
input = sys.stdin.readline

cList = ['c=', 'c-', 'dz=', 'd-', 'lj', 'nj', 's=', 'z='] # 크로아티아 알파벳 리스트
s = input().rstrip() # rstrip: 개행 문자 제거
cntList = [s.count(i) for i in cList] # 크로아티아 알파벳 세서 리스트로 저장

cntList[7] -= cntList[2] # dz= 와 z= 는 글자가 겹치므로 중복 제거

for i, count in enumerate(cntList): # 크로아티아 알파벳 제거
    for c in range(count):
        s = s.replace(cList[i], '')

cnt = sum(cntList) # 크로아티아 알파벳 개수 세는 변수

print(cnt+len(s))

 

다른 코드를 참고해봤는데, 복잡하게 하지 않아도 바로 구할 수 있는 코드가 있었다.

 

정답 코드

import sys
input = sys.stdin.readline

cList = ['c=', 'c-', 'dz=', 'd-', 'lj', 'nj', 's=', 'z='] # 크로아티아 알파벳 리스트
s = input().rstrip() # rstrip: 개행 문자 제거

for i in cList:
    s = s.replace(i, '1') # 크로아티아 알파벳을 1로 치환

print(len(s))

이런 아이디어는 어떻게 생각하는 걸까.. 아직 갈길이 멀다.

 

알게 된 점

• replace 사용법