結果

問題 No.2674 k-Walk on Bipartite
ユーザー ngtkana
提出日時 2024-03-02 03:36:44
言語 Rust
(1.83.0 + proconio)
結果
RE  
実行時間 -
コード長 3,136 bytes
コンパイル時間 12,399 ms
コンパイル使用メモリ 390,484 KB
実行使用メモリ 15,952 KB
最終ジャッジ日時 2024-09-29 22:59:40
合計ジャッジ時間 13,201 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28 RE * 8
権限があれば一括ダウンロードができます

ソースコード

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];
    assert!(start != end || !g[start].is_empty());
    let ans = match () {
        () if d == !0 => {
            if n == 2 && k % 2 == 0 {
                No
            } else {
                Unknown
            }
        }
        () if d % 2 != k % 2 => No,
        () if k < d => Unknown,
        () => {
            if start == end && g[start].is_empty() {
                Unknown
            } else {
                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