結果

問題 No.197 手品
ユーザー sinosino
提出日時 2020-04-17 11:40:17
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 1 ms / 1,000 ms
コード長 2,199 bytes
コンパイル時間 14,013 ms
コンパイル使用メモリ 378,520 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 04:01:49
合計ジャッジ時間 15,440 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff #

#![allow(unused_imports)]
#![allow(non_snake_case)]
use std::collections::HashMap;
use std::collections::HashSet;
use std::collections::VecDeque;

#[allow(unused_macros)]
macro_rules! read {
    ([$t:ty] ; $n:expr) =>
        ((0..$n).map(|_| read!([$t])).collect::<Vec<_>>());
    ($($t:ty),+ ; $n:expr) =>
        ((0..$n).map(|_| read!($($t),+)).collect::<Vec<_>>());
    ([$t:ty]) =>
        (rl().split_whitespace().map(|w| w.parse().unwrap()).collect::<Vec<$t>>());
    ($t:ty) =>
        (rl().parse::<$t>().unwrap());
    ($($t:ty),*) => {{
        let buf = rl();
        let mut w = buf.split_whitespace();
        ($(w.next().unwrap().parse::<$t>().unwrap()),*)
    }};
}

#[allow(dead_code)]
fn rl() -> String {
    let mut buf = String::new();
    std::io::stdin().read_line(&mut buf).unwrap();
    buf.trim_end().to_owned()
}

fn main() {
    const SUCCESS: &str = "SUCCESS";
    const FAILURE: &str = "FAILURE";

    let s = read!(String);
    let n = read!(u32);
    let e = read!(String);

    let s = s.into_bytes();
    let s = [s[0] == 'o' as u8, s[1] == 'o' as u8, s[2] == 'o' as u8];

    let e = e.into_bytes();
    let e = [e[0] == 'o' as u8, e[1] == 'o' as u8, e[2] == 'o' as u8];

    let msg = if s.iter().filter(|c| **c).count() != e.iter().filter(|c| **c).count() {
        SUCCESS
    } else {
        let mut set0 = HashSet::new();
        set0.insert(s.clone());
        let mut set1 = HashSet::new();
        set1.insert([s[0], s[2], s[1]]);
        set1.insert([s[1], s[0], s[2]]);
        let mut set2 = HashSet::new();
        set2.insert([s[2], s[0], s[1]]);
        set2.insert([s[1], s[2], s[0]]);
        let mut set3 = HashSet::new();
        set3.insert([s[2], s[1], s[0]]);

        let mut is_success = true;
        if n % 2 == 0 {
            is_success &= !set0.contains(&e);
            if n >= 2 {
                is_success &= !set2.contains(&e);
            }
        } else {
            is_success &= !set1.contains(&e);
            if n >= 3 {
                is_success &= !set3.contains(&e);
            }
        }

        if is_success {
            SUCCESS
        } else {
            FAILURE
        }
    };

    println!("{}", msg);
}
0