Study/Baekjoon

[Python] 백준 2941번 크로아티아 알파벳 | 문자열 내 특정 문자 찾기, str.replace()

yeorii 2023. 3. 28. 11:32

크로아티아 알파벳

문제

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

크로아티아 알파벳변경
č c=
ć c-
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

예제 입력 3 복사

nljj

예제 출력 3 복사

3

예제 입력 5 복사

dz=ak

예제 출력 5 복사

3

내 풀이

inp = input()
cro = ['c=', 'c-', 'dz=', 'd-', 'lj', 'nj', 's=', 'z=']
cnt = 0
for i in cro:
    if i in inp:
        cnt += inp.count(i)
print(len(inp)-cnt)

input 문자열에서 중복되게 세어지는 크로아티아 문자 수를 빼줬다.


str.replace(a, b) : 문자열에서 특정 문자 제거 및 변경 함수

풀이에는 쓰이지 않았지만 처음에 잘못 시도하면서 알게 된 함수 str.replace(a, b) 를 정리하겠다!

str.replace('c=', '') # 'c=' 제거
str.replace('c=', 'c-') # 'c='을 'c-'로 변경

str.replace(a, b) 이 함수는 문자열 내의 특정 문자인 a를 b로 바꿔준다.
이 함수를 사용해서 문제를 풀려고 했는데, 'nljj'같은 경우는 중간에 'lj'가 제거되면서 'nj'가 (세어지면 안될 것이) 세어져서 틀렸었다.

위 함수를 사용해서 푼 다른 풀이

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

'nj'가 안 세어질 수 있게 중간에 * 이나 다른 문자를 넣어두어서 하나로 세어지도록 하면 된다.