結果

問題 No.3463 Beltway
コンテスト
ユーザー QiToY
提出日時 2026-02-28 15:55:51
言語 Rust
(1.93.0 + proconio + num + itertools)
コンパイル:
/usr/bin/rustc_custom
実行:
./target/release/main
結果
WA  
実行時間 -
コード長 2,567 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,498 ms
コンパイル使用メモリ 209,376 KB
実行使用メモリ 55,272 KB
最終ジャッジ日時 2026-02-28 15:55:59
合計ジャッジ時間 5,902 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 16 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#![allow(unused_imports)]

fn main() {
    input! {
        n: usize, e: usize, f: Usize1, t: Usize1,
        e: [(Usize1, Usize1); e],
    }
    let mut g = vec![BTreeSet::new(); n];
    for (a, b) in e {
        g[a].insert(b);
        g[b].insert(a);
    }
    let mut loope = BTreeSet::new();
    {
        let mut g = g.clone();
        let mut que = VecDeque::new();
        for (v, g) in g.iter().enumerate() {
            if g.len() == 1 {
                que.push_back(v);
            }
        }
        while let Some(v) = que.pop_front() {
            let e = g[v].clone();
            for u in e {
                g[u].remove(&v);
                g[v].remove(&u);
                if g[u].len() == 1 {
                    que.push_back(u);
                }
            }
        }
        for (v, g) in g.into_iter().enumerate() {
            for u in g {
                loope.insert((v, u));
            }
        }
    }
    // eprintln!("{loope:?}");
    let mut que = VecDeque::new();
    let mut dp = vec![(usize::MAX, 0); n];
    que.push_back(f);
    dp[f] = (0, 0);
    while let Some(v) = que.pop_front() {
        for &u in &g[v] {
            match dp[u].0.cmp(&(dp[v].0 + 1)) {
                std::cmp::Ordering::Less => { }
                std::cmp::Ordering::Equal => {
                    chmax!(dp[u].1, dp[v].1 + if loope.contains(&(v, u)) { 1 } else { 0 });
                }
                std::cmp::Ordering::Greater => {
                    dp[u] = dp[v];
                    dp[u].0 += 1;
                    if loope.contains(&(v, u)) {
                        dp[u].1 += 1;
                    }
                    que.push_back(u);
                }
            }
        }
    }
    if dp[t].0 == usize::MAX {
        println!("-1");
    } else {
        println!("{}", dp[t].1);
    }
}

use proconio::{input, marker::*};
use itertools::{iproduct, izip, Itertools as _};
use std::{cmp::Reverse, collections::*};

#[macro_export]
macro_rules! chmax {
    ($a:expr, $b:expr) => {{
        let tmp = $b;
        if $a < tmp {
            $a = tmp;
            true
        } else {
            false
        }
    }};
}

#[macro_export]
macro_rules! chmin {
    ($a:expr, $b:expr) => {{
        let tmp = $b;
        if $a > tmp {
            $a = tmp;
            true
        } else {
            false
        }
    }};
}

#[macro_export]
/// mvec![]
macro_rules! mvec {
    ($val:expr; ()) => {
        $val
    };
    ($val:expr; ($size:expr $(,$rest:expr)*)) => {
        vec![mvec![$val; ($($rest),*)]; $size]
    };
}
0