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

[Silver V] 크로아티아 알파벳 - 2941

Posted by Hebi on April 6, 2023

백준 알고리즘

[Silver V] 크로아티아 알파벳 - 2941

문제 링크

성능 요약

메모리: 13156 KB, 시간: 4 ms

분류

구현(implementation), 문자열(string)

문제 설명

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

크로아티아 알파벳 변경
č c=
ć c-
dz=
đ d-
lj lj
nj nj
š s=
ž z=

예를 들어, ljes=njak은 크로아티아 알파벳 6개(lj, e, š, nj, a, k)로 이루어져 있다. 단어가 주어졌을 때, 몇 개의 크로아티아 알파벳으로 이루어져 있는지 출력한다.

dž는 무조건 하나의 알파벳으로 쓰이고, d와 ž가 분리된 것으로 보지 않는다. lj와 nj도 마찬가지이다. 위 목록에 없는 알파벳은 한 글자씩 센다.

입력

첫째 줄에 최대 100글자의 단어가 주어진다. 알파벳 소문자와 '-', '='로만 이루어져 있다.

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

출력

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

Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
use std::collections::HashSet;
use std::io;

fn main() {
    let mut croatia = String::new();
    io::stdin()
        .read_line(&mut croatia)
        .expect("Failed to read input");
    croatia = croatia.trim().to_string();

    let word_map: HashSet<&str> = ["c=", "c-", "d-", "lj", "nj", "s=", "z="]
        .iter()
        .cloned()
        .collect();

    let mut total = 1;
    for i in 1..croatia.len() {
        let mut temp: String = croatia[i - 1..i + 1].into();
        if word_map.contains(&*temp) {
            if temp == "z=" && i > 1 {
                let temp: String = croatia[i - 2..i + 1].into();
                if temp == "dz=" {
                    total -= 1;
                }
            }

            continue;
        }

        total += 1;
    }

    println!("{}", total);
}