백준 2941 - 크로아티아 알파벳 (파이썬)

문제

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

크로아티아 알파벳변경

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

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

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

입력

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

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

출력

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

# 해결법

 

0에서 문자열의 길이만큼 반복해준다.

문자가 2개보다 작으면 문자열의 길이를 출력한다.

문자가 2개일때 크로아티아 알파벳이라면 1을 출력한다.

연달아 있는 문자가 3개를 더했을 때 크로아티아 알파벳이라면 길이에서 -2를 해준다.

연달아 있는 문자가 2개를 더했을 때 크로아티아 알파벳이라면 길이에서 -1을 해준다.

 

# 나의 코드

x = input()
y = len(x)
cro_al = ['c=','c-','dz=','d-','lj','nj','s=','z=']
for i in range(0, len(x)):
    if len(x) < 2:
        y = len(x)
    elif len(x) == 2:
        if (x[i-2] + x[i-1]) in cro_al:
            y = 1
    elif (x[i-3]+x[i-2]+x[i-1]) in cro_al:
        y -= 2
    elif (x[i-2] + x[i-1]) in cro_al:
        y -= 1
print(y)

 

# 개선 코드

 

검색을 해보니 크로아티아 알파벳이 입력값에 있다면 그냥 한글자 알파벳으로 바꿔주는 방법이 있었다.

치환을 끝낸 후 길이를 리턴하는 방법으로 구현해보았다. 

구현할때 dz= 가 z= 보다 앞에 있지 않으면 z=를 우선으로 바꿔버리게되므로 dz=가 z= 보다 앞에 있어야 한다.

x = input()
cro_al = ['c=','c-','dz=','d-','lj','nj','s=','z=']
for i in cro_al:
    x = x.replace(i, 'a')
print(len(x))