結果

問題 No.3291 K-step Navigation
ユーザー koba-e964
提出日時 2025-09-18 10:24:19
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 2,507 bytes
コンパイル時間 12,265 ms
コンパイル使用メモリ 399,292 KB
実行使用メモリ 7,716 KB
最終ジャッジ日時 2025-09-18 10:24:35
合計ジャッジ時間 15,087 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 47 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::*;
// https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
macro_rules! input {
    ($($r:tt)*) => {
        let stdin = std::io::stdin();
        let mut bytes = std::io::Read::bytes(std::io::BufReader::new(stdin.lock()));
        let mut next = move || -> String{
            bytes.by_ref().map(|r|r.unwrap() as char)
                .skip_while(|c|c.is_whitespace())
                .take_while(|c|!c.is_whitespace())
                .collect()
        };
        input_inner!{next, $($r)*}
    };
}

macro_rules! input_inner {
    ($next:expr) => {};
    ($next:expr,) => {};
    ($next:expr, $var:ident : $t:tt $($r:tt)*) => {
        let $var = read_value!($next, $t);
        input_inner!{$next $($r)*}
    };
}

macro_rules! read_value {
    ($next:expr, ( $($t:tt),* )) => { ($(read_value!($next, $t)),*) };
    ($next:expr, [ $t:tt ; $len:expr ]) => {
        (0..$len).map(|_| read_value!($next, $t)).collect::<Vec<_>>()
    };
    ($next:expr, usize1) => (read_value!($next, usize) - 1);
    ($next:expr, $t:ty) => ($next().parse::<$t>().expect("Parse error"));
}


fn find_shortest_odd_cycle(g: &[Vec<usize>], s: usize) -> Option<i32> {
    const INF: i32 = 1 << 28;
    let n = g.len() / 2;
    let mut dist = vec![INF; 2 * n];
    let mut que = VecDeque::new();
    que.push_back((0, s));
    while let Some((d, v)) = que.pop_front() {
        if dist[v] <= d { continue; }
        dist[v] = d;
        for &w in &g[v] {
            que.push_back((d + 1, w));
        }
    }
    if dist[s + n] == INF {
        None
    } else {
        Some(dist[s + n])
    }
}

fn main() {
    input! {
        n: usize, m: usize, k: i64, s: usize1, t: usize1,
        uv: [(usize1, usize1); m],
    }
    if k % 2 == 1 {
        println!("Yes");
        return;
    }
    let mut g = vec![vec![]; 2 * n];
    let mut has_st = false;
    for &(u, v) in &uv {
        if (u, v) == (s, t) || (u, v) == (t, s) {
            has_st = true;
            continue;
        }
        g[u].push(v + n);
        g[u + n].push(v);
        g[v + n].push(u);
        g[v].push(u + n);
    }
    if !has_st {
        println!("{}", if g[s].is_empty() && g[t].is_empty() { "No" } else { "Yes" });
        return;
    }
    let mut dist = k + 2;
    for i in 0..n {
        if let Some(d) = find_shortest_odd_cycle(&g, i) {
            dist = dist.min(d as i64);
        }
    }
    if dist <= k { // <- WA! `k` should be `k - 3`
        println!("Yes");
    } else {
        println!("No");
    }
}
0