Add a test for ABC120-D

Author: qryxip

examples/abc120-d.rs

@@ -0,0 +1,29 @@
+// https://atcoder.jp/contests/abc120/tasks/abc120_d
+
+use proconio::marker::Usize1;
+use proconio::{fastout, input};
+use union_find::{QuickFindUf, UnionBySize, UnionFind as _};
+
+#[fastout]
+fn main() {
+    input! {
+        n: usize,
+        m: usize,
+        abs: [(Usize1, Usize1); m],
+    }
+
+    let mut u = QuickFindUf::<UnionBySize>::new(n);
+    let mut k = n * (n - 1) / 2;
+    let mut r = vec![k];
+    r.extend(abs.into_iter().rev().map(|(a, b)| {
+        let p = u.get(a).size() * u.get(b).size();
+        if u.union(a, b) {
+            k -= p;
+        }
+        k
+    }));
+    assert_eq!(r.pop(), Some(0));
+    for r in r.into_iter().rev() {
+        println!("{}", r);
+    }
+}

examples/tests.ron

@@ -12,5 +12,9 @@
             name: "ABC049 / ARC065: C - 白昼夢 / Daydream",
             matching: ExactWords,
         ),
+        "abc120-d": (
+            name: "ABC120: D - Decayed Bridges ",
+            matching: ExactWords,
+        ),
     }
 )

examples/testsets/abc120-d/in/sample_01.txt

@@ -0,0 +1,6 @@
+4 5
+1 2
+3 4
+1 3
+2 3
+1 4

examples/testsets/abc120-d/in/sample_02.txt

@@ -0,0 +1,6 @@
+6 5
+2 3
+1 2
+5 6
+3 4
+4 5

examples/testsets/abc120-d/in/sample_03.txt

@@ -0,0 +1,2 @@
+2 1
+1 2

examples/testsets/abc120-d/out/sample_01.txt

@@ -0,0 +1,5 @@
+0
+0
+4
+5
+6

examples/testsets/abc120-d/out/sample_02.txt

@@ -0,0 +1,5 @@
+8
+9
+12
+14
+15

examples/testsets/abc120-d/out/sample_03.txt

@@ -0,0 +1 @@
+1