library_checker/data_structure/unionfind.rs
1use competitive::data_structure::UnionFind;
2use competitive::prelude::*;
3
4competitive::define_enum_scan! {
5 enum Query: usize {
6 0 => Unite { u: usize, v: usize }
7 1 => Same { u: usize, v: usize }
8 }
9}
10
11#[verify::library_checker("unionfind")]
12pub fn unionfind(reader: impl Read, writer: impl Write) {
13 prepare_io!(reader, writer);
14 sc!(n, q);
15 let mut uf = UnionFind::new(n);
16 for _ in 0..q {
17 sc!(query: Query);
18 match query {
19 Query::Unite { u, v } => {
20 uf.unite(u, v);
21 }
22 Query::Same { u, v } => {
23 pp!(uf.same(u, v) as usize);
24 }
25 }
26 }
27}