백준 알고리즘

[Silver IV] 붙임성 좋은 총총이 - 26069

문제 링크

성능 요약

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

분류

자료 구조, 해시를 사용한 집합과 맵, 트리를 사용한 집합과 맵

문제 설명

무지개 댄스를 추는 총총과 그걸 보는 총총

총총이는 친구 곰곰이의 소개로 제2회 곰곰컵에 출연할 기회를 얻었다!

총총이는 자신의 묘기인 무지개 댄스를 선보여, 여러분의 환심을 사려 한다. 이 댄스는 중독성이 강하기 때문에, 한번 보게 된 사람은 모두 따라 하게 돼버린다.

무지개 댄스를 추는 총총 2마리

사람들이 만난 기록이 시간 순서대로 $N$ 개 주어진다. (총총이는 토끼이지만 이 문제에서는 편의상 사람이라고 가정한다.)

무지개 댄스를 추지 않고 있던 사람이 무지개 댄스를 추고 있던 사람을 만나게 된다면, 만난 시점 이후로 무지개 댄스를 추게 된다.

기록이 시작되기 이전 무지개 댄스를 추고 있는 사람은 총총이 뿐이라고 할 때, 마지막 기록 이후 무지개 댄스를 추는 사람이 몇 명인지 구해보자!

입력

첫번째 줄에는 사람들이 만난 기록의 수 $N (1 \leq N \leq 1 000)$ 이 주어진다.

두번째 줄부터 $N$ 개의 줄에 걸쳐 사람들이 만난 기록이 주어진다. $i + 1$ 번째 줄에는 $i$ 번째로 만난 사람들의 이름 $A_{i}$ 와 $B_{i}$ 가 공백을 사이에 두고 주어진다. $A_{i}$ 와 $B_{i}$ 는 숫자와 영문 대소문자로 이루어진 최대 길이 $20$ 의 문자열이며, 서로 같지 않다.

총총이의 이름은 ChongChong으로 주어지며, 기록에서 1회 이상 주어진다.

동명이인은 없으며, 사람의 이름은 대소문자를 구분한다. (ChongChong과 chongchong은 다른 이름이다.)

출력

마지막 기록 이후 무지개 댄스를 추는 사람의 수를 출력하라.

Code

use std::collections::HashMap;
use std::io::{self, BufRead};

fn main() -> io::Result<()> {
    let stdin = io::stdin();
    let mut handle = stdin.lock();

    let mut dance: HashMap<String, bool> = HashMap::new();
    dance.insert("ChongChong".to_string(), true);
    let mut answer = 1;

    let mut input = String::new();
    handle.read_line(&mut input)?;
    let n: usize = input.trim().parse().unwrap();

    for _ in 0..n {
        input.clear();
        handle.read_line(&mut input)?;
        let words: Vec<String> = input.split_whitespace().map(|s| s.to_string()).collect();
        let a = &words[0];
        let b = &words[1];

        if dance.contains_key(a) {
            if !dance.contains_key(b) {
                dance.insert(b.to_string(), true);
                answer += 1;
            }
        } else if dance.contains_key(b) {
            dance.insert(a.to_string(), true);
            answer += 1;
        }
    }

    println!("{}", answer);

    Ok(())
}

백준 26069 - 붙임성 좋은 총총이 [Rust]