aizu_online_judge/dsl/
dsl_1_b.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{algebra::AdditiveOperation, data_structure::PotentializedUnionFind};
4
5competitive::define_enum_scan! {
6    enum Query: usize {
7        0 => Unite { x: usize, y: usize, w: i64 }
8        1 => Diff { x: usize, y: usize }
9    }
10}
11
12#[verify::aizu_online_judge("DSL_1_B")]
13pub fn dsl_1_b(reader: impl Read, mut writer: impl Write) {
14    let s = read_all_unchecked(reader);
15    let mut scanner = Scanner::new(&s);
16    scan!(scanner, n, q);
17    let mut uf = PotentializedUnionFind::<AdditiveOperation<_>>::new(n);
18    for _ in 0..q {
19        scan!(scanner, query: Query);
20        match query {
21            Query::Unite { x, y, w } => {
22                uf.unite_with(x, y, w);
23            }
24            Query::Diff { x, y } => {
25                if let Some(w) = uf.difference(x, y) {
26                    writeln!(writer, "{}", w).ok();
27                } else {
28                    writeln!(writer, "?").ok();
29                }
30            }
31        }
32    }
33}
aizu_online_judge/dsl/dsl_1_b.rs

aizu_online_judge/dsl/
dsl_1_b.rs
