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

🔗 문제 링크

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

문제

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

크로아티아 알파벳 변경

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

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

입력

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

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

출력

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

💻 코드

import sys 
input = sys.stdin.readline 

cro = ['c=', 'c-', 'dz=', 'd-', 'lj', 'nj', 's=', 'z=']
res = 0

cur = 0
s = input().rstrip()
while cur < len(s):   

    if s[cur] in 'cdlnsz':
        if cur < len(s)-1:
            if s[cur] =='c' and (s[cur+1] == '=' or s[cur+1] == '-'):
                res += 1
                cur += 2 
                continue
            elif s[cur] == 'd':
                if cur < len(s)-2 and s[cur+1] == 'z' and s[cur+2] == '=':
                    res += 1
                    cur += 3
                    continue
                elif s[cur+1] == '-':
                    res += 1
                    cur += 2
                    continue
                else:
                    res += 1
            elif s[cur] == 'l' and s[cur+1] == 'j':
                res += 1
                cur += 2
                continue
            elif s[cur] == 'n' and s[cur+1] == 'j':
                res += 1
                cur += 2
                continue
            elif s[cur] =='s' and s[cur+1] == '=':
                res += 1
                cur += 2
                continue
            elif s[cur] =='z' and s[cur+1] == '=':
                res += 1
                cur += 2
                continue
            else:
                res += 1

        else:
            res += 1
    else:
        res += 1

    cur += 1
        
print(res)

✨ 코드 개선

import sys 
input = sys.stdin.readline 
s = input().rstrip()
for p in ['dz=', 'c=', 'c-', 'd-', 'lj', 'nj', 's=', 'z=']:
    s = s.replace(p, '#')
print(len(s))

• replace로 긴 패턴부터 1글자로 치환한 뒤 len()만 세면 끝