Add a test for ATC001-B

qryxip · qryxip · commit 794585af06d8 · 2019-12-13T12:50:17.000+09:00
diff --git a/examples/atc001-b.rs b/examples/atc001-b.rs
@@ -0,0 +1,48 @@
+use petgraph::unionfind::UnionFind;
+
+#[proconio::fastout]
+fn main() {
+    // use std::io::{self, Read as _};
+    //
+    // let mut input = "".to_owned();
+    // io::stdin().read_to_string(&mut input).unwrap();
+    // let mut input = input.split_whitespace();
+    // macro_rules! read {
+    //     ([$t:tt; $n:expr]) => {
+    //         (0..$n).map(|_| read!($t)).collect::<Vec<_>>()
+    //     };
+    //     (($($t:tt),+)) => {
+    //         ($(read!($t)),*)
+    //     };
+    //     (_1based) => {
+    //         read!(usize) - 1
+    //     };
+    //     (_bytes) => {
+    //         read!(String).into_bytes()
+    //     };
+    //     ($ty:ty) => {
+    //         input.next().unwrap().parse::<$ty>().unwrap()
+    //     };
+    // }
+    //
+    // let (n, q) = read!((usize, usize));
+    // let pabs = read!([(u8, usize, usize); q]);
+
+    use proconio::input;
+
+    input! {
+        n: usize,
+        q: usize,
+        pabs: [(u8, usize, usize); q],
+    }
+
+    let mut uf = UnionFind::new(n);
+    for (p, a, b) in pabs {
+        if p == 1 {
+            let same = uf.find(a) == uf.find(b);
+            println!("{}", if same { "Yes" } else { "No" });
+        } else {
+            uf.union(a, b);
+        }
+    }
+}
diff --git a/examples/tests.ron b/examples/tests.ron
@@ -8,6 +8,10 @@
             name: "practice contest: A - Welcome to AtCoder",
             matching: ExactWords,
         ),
+        "atc001-b": (
+            name: "ATC001: B - Union Find",
+            matching: ExactWords,
+        ),
         "arc065-c": (
             name: "ABC049 / ARC065: C - 白昼夢 / Daydream",
             matching: ExactWords,
diff --git a/examples/testsets/atc001-b/in/00_sample_01.txt b/examples/testsets/atc001-b/in/00_sample_01.txt
@@ -0,0 +1,10 @@
+8 9
+0 1 2
+0 3 2
+1 1 3
+1 1 4
+0 2 4
+1 4 1
+0 4 2
+0 0 0
+1 0 0
diff --git a/examples/testsets/atc001-b/out/00_sample_01.txt b/examples/testsets/atc001-b/out/00_sample_01.txt
@@ -0,0 +1,4 @@
+Yes
+No
+Yes
+Yes

-Original file line number
+Diff line change
@@ @@ -0,0 +1,10 @@ @@
 +8 9
 +0 1 2
 +0 3 2
 +1 1 3
 +1 1 4
 +0 2 4
 +1 4 1
 +0 4 2
 +0 0 0
 +1 0 0