結果

問題 No.2674 k-Walk on Bipartite
ユーザー ngtkana
提出日時 2024-03-02 03:27:55
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 3,052 bytes
コンパイル時間 13,198 ms
コンパイル使用メモリ 390,144 KB
実行使用メモリ 15,804 KB
最終ジャッジ日時 2024-09-29 22:58:20
合計ジャッジ時間 13,346 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

use input::input_array;
use std::collections::VecDeque;
use Ans::*;

fn main() {
    let [n, m] = input_array::<usize, 2>();
    let [start, end, k] = input_array::<usize, 3>();
    let start = start - 1;
    let end = end - 1;
    let mut g = vec![Vec::new(); n];
    for _ in 0..m {
        let [i, j] = input_array::<usize, 2>();
        let i = i - 1;
        let j = j - 1;
        g[i].push(j);
        g[j].push(i);
    }
    let mut dist = vec![std::usize::MAX; g.len()];
    dist[start] = 0;
    let mut queue = VecDeque::from(vec![start]);
    while let Some(x) = queue.pop_front() {
        for &y in &g[x] {
            if dist[y] == std::usize::MAX {
                dist[y] = dist[x] + 1;
                queue.push_back(y);
            }
        }
    }
    let d = dist[end];
    let ans = match n {
        2 => match () {
            () if usize::from(start == end) != (k % 2) => No,
            () if (start == end || m == 1) => Yes,
            () => Unknown,
        },
        _ => match () {
            () if d == !0 => Unknown,
            () if d % 2 != k % 2 => No,
            () if k < d => Unknown,
            () => Yes,
        },
    };
    println!(
        "{}",
        match ans {
            Yes => "Yes",
            No => "No",
            Unknown => "Unknown",
        }
    );
}

#[derive(Clone, Copy)]
enum Ans {
    Yes,
    No,
    Unknown,
}

// input {{{
#[allow(dead_code)]
mod input {
    use std::cell::Cell;
    use std::convert::TryFrom;
    use std::io::stdin;
    use std::io::BufRead;
    use std::io::BufReader;
    use std::io::Lines;
    use std::io::Stdin;
    use std::str::FromStr;
    use std::sync::Mutex;
    use std::sync::Once;
    type Server = Mutex<Lines<BufReader<Stdin>>>;
    static ONCE: Once = Once::new();
    pub struct Lazy(Cell<Option<Server>>);
    unsafe impl Sync for Lazy {}
    fn line() -> String {
        static SYNCER: Lazy = Lazy(Cell::new(None));
        ONCE.call_once(|| {
            SYNCER
                .0
                .set(Some(Mutex::new(BufReader::new(stdin()).lines())));
        });
        unsafe {
            (*SYNCER.0.as_ptr())
                .as_ref()
                .unwrap()
                .lock()
                .unwrap()
                .next()
                .unwrap()
                .unwrap()
        }
    }
    pub trait ForceFromStr: FromStr {
        fn force_from_str(s: &str) -> Self;
    }
    impl<T, E> ForceFromStr for T
    where
        T: FromStr<Err = E>,
        E: std::fmt::Debug,
    {
        fn force_from_str(s: &str) -> Self {
            s.parse().unwrap()
        }
    }
    pub fn input_array<T: ForceFromStr, const N: usize>() -> [T; N]
    where
        T: std::fmt::Debug,
    {
        <[_; N]>::try_from(input_vec()).unwrap()
    }
    pub fn input_vec<T: ForceFromStr>() -> Vec<T> {
        line()
            .split_whitespace()
            .map(T::force_from_str)
            .collect::<Vec<_>>()
    }
    pub fn input<T: ForceFromStr>() -> T {
        T::force_from_str(&line())
    }
}
// }}}
0