[Rust로 백준 하루 하나] 6-6. 크로아티아 알파벳

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

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

출력

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

---

풀이

코드

use std::io;

fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    input = input.trim().to_string();
    
    let croatia_alphabet = vec!["c=", "c-", "dz=", "d-", "lj", "nj", "s=", "z="];
    let mut index = 0;
    let mut count = 0;
    
    while index < input.len() {
        let mut matched = false;
    
        for &alphabet in &croatia_alphabet {
            if index+alphabet.len() <= input.len() && &input[index..index+alphabet.len()] == alphabet {
                count += 1;
                index += alphabet.len();
                matched = true;
                break;
            }
        }
        
        if !matched {
            count += 1;
            index += 1;
        }
    }
    
    println!("{}", count);
}

해설

문자열을 앞에서 부터 길이를 세면서 읽어야 중복되는 알파벳을 다시 세지 않을 수 있다. 문제 풀이 방법을 잘 알아두도록 하자.

if 문에서 && 같은 다중 조건의 경우 앞 조건을 먼저 검사하고 뒤의 조건을 검사하기 때문에 먼저 검사할 조건을 잘 고려해야 한다.

---

추가 학습

• 특이사항 없음