aizu_online_judge/dsl/
dsl_1_b.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{algebra::AdditiveOperation, data_structure::PotentializedUnionFind};
4
5#[verify::aizu_online_judge("DSL_1_B")]
6pub fn dsl_1_b(reader: impl Read, mut writer: impl Write) {
7    let s = read_all_unchecked(reader);
8    let mut scanner = Scanner::new(&s);
9    scan!(scanner, n, q);
10    let mut uf = PotentializedUnionFind::<AdditiveOperation<_>>::new(n);
11    for _ in 0..q {
12        scan!(scanner, ty, x, y);
13        if ty == 0 {
14            scan!(scanner, w: i64);
15            uf.unite_with(x, y, w);
16        } else if let Some(w) = uf.difference(x, y) {
17            writeln!(writer, "{}", w).ok();
18        } else {
19            writeln!(writer, "?").ok();
20        }
21    }
22}