結果

問題 No.1468 Colourful Pastel
ユーザー Masaki Kitaguchi
提出日時 2022-01-09 23:47:44
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 29 ms / 1,000 ms
コード長 1,518 bytes
コンパイル時間 14,189 ms
コンパイル使用メモリ 403,444 KB
実行使用メモリ 14,208 KB
最終ジャッジ日時 2024-11-14 10:37:44
合計ジャッジ時間 14,411 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#[macro_export]
macro_rules! setup {
    { mut $input:ident: SplitWhitespace $(,)? } => {
        use std::io::Read;
        let mut buf = String::new();
        std::io::stdin().read_to_string(&mut buf).unwrap();
        let mut $input = buf.split_whitespace();
    };
}

#[macro_export]
macro_rules! parse_next {
    ($str_iter:expr) => {
        $str_iter.next().unwrap().parse().unwrap()
    };
}

use std::collections::HashMap;

fn main() {
    setup! { mut input: SplitWhitespace };

    let n: usize = parse_next!(input);
    let a: Vec<String> = (0..n).map(|_| parse_next!(input)).collect();
    let b: Vec<String> = (0..(n - 1)).map(|_| parse_next!(input)).collect();

    let mut colours = HashMap::new();
    [
        ("Red", 0),
        ("Orange", 1),
        ("Yellow", 2),
        ("Green", 3),
        ("Cyan", 4),
        ("Blue", 5),
        ("Violet", 6),
    ]
    .iter()
    .for_each(|(s, idx)| {
        colours.insert(String::from(*s), *idx as usize);
    });

    let mut count_a = vec![0; 7];
    for i in 0..n {
        count_a[*colours.get(&a[i]).unwrap()] += 1;
    }

    let mut count_b = vec![0; 7];
    for i in 0..(n - 1) {
        count_b[*colours.get(&b[i]).unwrap()] += 1;
    }

    for j in 0..7 {
        if count_a[j] != count_b[j] {
            let colour = colours
                .iter()
                .filter(|(_, idx)| **idx == j)
                .map(|(s, _)| s)
                .nth(0)
                .unwrap();
            println!("{}", colour);
        }
    }
}
0