[Python] 백준 2941번 - 크로아티아 알파벳

📖 문제

2941번: 크로아티아 알파벳

 

🔎 풀이

어떻게 푸느냐에 따라 코드의 간결성이 크게 차이나는 문제입니다. 주어진 문자열의 크기가 크지 않기 때문에 count 함수를 이용하여 크로아티아 알파벳들을 하나씩 세줍니다. 이때 전체 문자열의 합을 구해놓고, 크로아티아 알파벳 개수만큼 빼주는 식으로 구현합니다. (크로아티아 알파벳들은 두 자리씩 차지하므로) 'dz='는 '=' 에서 한 번 더 걸러지므로 총 2번을 빼, 알맞은 결과가 출력됩니다.

 

💻 코드

arr = input()
print(len(arr)-sum(arr.count(x) for x in ['=', '-', 'lj', 'nj', 'dz=']))

 

물론 하드코딩으로도 풀 수 있습니다..

뒤에서부터 인덱스로 조회하며, 조건에 따라 합해줍니다.

import sys
input = sys.stdin.readline

arr = input().strip()
croa = ['c=', 'c-', 'dz=', 'd-', 'lj', 'nj', 's=', 'z=']
ans = 0
i = len(arr) - 1
while i >= 0:
    if arr[i] == '-':
        ans += 1
        i -= 2
    elif arr[i] == '=':
        if i >= 2 and arr[i-2:i] == 'dz':
            ans += 1
            i -= 3
        else:
            ans += 1
            i -= 2
    elif i >= 1 and (arr[i-1:i+1] == 'nj' or arr[i-1:i+1] == 'lj'):
        ans += 1
        i -=2
    else:
        ans += 1
        i -= 1
    
print(ans)