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

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

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글자의 단어가 주어진다. 알파벳 소문자와 '-', '='로만 이루어져 있다.

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

출력

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

예제 입력 1

ljes=njak

예제 출력 1

6

예제 입력 2

ddz=z=

예제 출력 2

3

예제 입력 3

nljj

예제 출력 3

3

예제 입력 4

c=c=

예제 출력 4

2

풀이 설명

1. 배열 arr에 문제의 크로아티아 알파벳을 넣은 후 시작한다.

2. arr을 탐색하며 문자열 s에 각 원소가 포함되어있는지 확인

  2-1. 만약 포함되어있다면 포함된 개수를 cnt에 더하고,

  2-2. s에서 해당 원소를 전부 " " 로 대체한다.

        (""로 바꾸면 바꾼 뒤의 앞뒤 문자열이 붙으면서 기존에 없던 크로아티아 알파벳이 생길 수가 있음

           ex. nljj에서 lj가 삭제되면 nj가 되는데, 이것은 크로아티아 알파벳이므로 잘못된 계산이 뒤따른다.)

3. 2번이 끝나면 대체했던 " "을 ""로 바꾸고 그 길이를 cnt에 더한다.

Python3 코드

s = input()
arr = ["c=","c-","dz=","d-","lj","nj","s=","z="]
cnt = 0

for i in arr:
    if i in s:
        cnt += s.count(i)
        s = s.replace(i, " ")
        #print(s)
        
cnt += len(s.replace(" ",""))
print(cnt)