BOJ: 크로아티아 알파벳 - 2941번

2941번: 크로아티아 알파벳

 

문제

예전에는 운영체제에서 크로아티아 알파벳을 입력할 수가 없었다. 따라서, 다음과 같이 크로아티아 알파벳을 변경해서 입력했다.
예를 들어, ljes=njak은 크로아티아 알파벳 6개(lj, e, š, nj, a, k)로 이루어져 있다. 단어가 주어졌을 때, 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.
dž는 무조건 하나의 알파벳으로 쓰이고, d와 ž가 분리된 것으로 보지 않는다. lj와 nj도 마찬가지이다. 위 목록에 없는 알파벳은 한 글자씩 센다.

 

내가 작성한 코드

# <https://www.acmicpc.net/problem/2941>
# 크로아티아 알파벳
# c=, c-, dz=, d-, lj, nj, s=, z=

import sys

text = sys.stdin.readline().strip()
i = 0
num = 0

if len(text) < 2:
    print(1)
else:
    while i < len(text):
        if i + 1 >= len(text):
            num += 1
            break
        if text[i] == "c" and text[i+1] == "=" or text[i+1] == "-":
            num += 1
            i += 2
            continue
        if text[i] == "d":
            if text[i+1] == "-":
                num += 1
                i += 2
                continue
            elif i+2 < len(text) and (text[i+1] == "z" and text[i+2] == "="):
                num += 1
                i += 3
                continue
        if text[i] == "l" and text[i+1] == "j":
            num += 1
            i += 2
            continue
        if text[i] == "n" and text[i+1] == "j":
            num += 1
            i += 2
            continue
        if text[i] == "s" and text[i+1] == "=":
            num += 1
            i += 2
            continue
        if text[i] == "z" and text[i+1] == "=":
            num += 1
            i += 2
            continue
        i += 1
        num += 1

    print(num)

 

• 어떻게 문자를 비교하는게 좋을 지 고민하다가, 가장 쉬운 if문을 떡칠하는 것으로(...) 풀었다.

• vscode에서 실행할 때는 정상 출력인데, 백준에서는 계속 런타임 에러가 떠서 구글링을 했다. 이유는, i+1이나 i+2의 위치한 문자가 없는 경우(입력받은 문자열의 길이보다 인덱스가 큰 경우)를 검사하지 않아서였다. 그래서 중간에 i+1, i+2를 검사하는 코드를 추가했다.

• lj, nj, s=, z=를 전부 한번에 검사하는 방식으로 아래와 같이 코드를 바꿔봤었는데, 코드 길이는 줄어들었지만 오히려 시간이 더 증가했다.

    if text[i] == "l" and text[i+1] == "j" or text[i] == "n" and text[i+1] == "j" or text[i] == "s" and text[i+1] == "=" or text[i] == "z" and text[i+1] == "=":
        num += 1
        i += 2
        continue