library_checker/data_structure/
unionfind.rs1#[doc(no_inline)]
2pub use competitive::data_structure::UnionFind;
3use competitive::prelude::*;
4
5competitive::define_enum_scan! {
6 enum Query: usize {
7 0 => Unite { u: usize, v: usize }
8 1 => Same { u: usize, v: usize }
9 }
10}
11
12#[verify::library_checker("unionfind")]
13pub fn unionfind(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 = UnionFind::new(n);
18 for _ in 0..q {
19 scan!(scanner, query: Query);
20 match query {
21 Query::Unite { u, v } => {
22 uf.unite(u, v);
23 }
24 Query::Same { u, v } => {
25 writeln!(writer, "{}", uf.same(u, v) as usize).ok();
26 }
27 }
28 }
29}