aizu_online_judge/dsl/
dsl_1_a.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 { x: usize, y: usize }
8 1 => Same { x: usize, y: usize }
9 }
10}
11
12#[verify::aizu_online_judge("DSL_1_A")]
13pub fn dsl_1_a(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 { x, y } => {
22 uf.unite(x, y);
23 }
24 Query::Same { x, y } => {
25 writeln!(writer, "{}", (uf.same(x, y) as usize)).ok();
26 }
27 }
28 }
29}