aizu_online_judge/dsl/dsl_1_a.rs
1use competitive::data_structure::UnionFind;
2use competitive::prelude::*;
3
4competitive::define_enum_scan! {
5 enum Query: usize {
6 0 => Unite { x: usize, y: usize }
7 1 => Same { x: usize, y: usize }
8 }
9}
10
11#[verify::aizu_online_judge("DSL_1_A")]
12pub fn dsl_1_a(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 { x, y } => {
20 uf.unite(x, y);
21 }
22 Query::Same { x, y } => {
23 pp!((uf.same(x, y) as usize));
24 }
25 }
26 }
27}