Add a test for ABC120-D

qryxip · qryxip · commit cdd5a5fa7fd7 · 2019-12-12T17:58:44.000+09:00
diff --git a/examples/abc120-d.rs b/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);
+    }
+}
diff --git a/examples/tests.ron b/examples/tests.ron
@@ -12,5 +12,9 @@
             name: "ABC049 / ARC065: C - 白昼夢 / Daydream",
             matching: ExactWords,
         ),
+        "abc120-d": (
+            name: "ABC120: D - Decayed Bridges ",
+            matching: ExactWords,
+        ),
     }
 )
diff --git a/examples/testsets/abc120-d/in/sample_01.txt b/examples/testsets/abc120-d/in/sample_01.txt
@@ -0,0 +1,6 @@
+4 5
+1 2
+3 4
+1 3
+2 3
+1 4
diff --git a/examples/testsets/abc120-d/in/sample_02.txt b/examples/testsets/abc120-d/in/sample_02.txt
@@ -0,0 +1,6 @@
+6 5
+2 3
+1 2
+5 6
+3 4
+4 5
diff --git a/examples/testsets/abc120-d/in/sample_03.txt b/examples/testsets/abc120-d/in/sample_03.txt
@@ -0,0 +1,2 @@
+2 1
+1 2
diff --git a/examples/testsets/abc120-d/out/sample_01.txt b/examples/testsets/abc120-d/out/sample_01.txt
@@ -0,0 +1,5 @@
+0
+0
+4
+5
+6
diff --git a/examples/testsets/abc120-d/out/sample_02.txt b/examples/testsets/abc120-d/out/sample_02.txt
@@ -0,0 +1,5 @@
+8
+9
+12
+14
+15
diff --git a/examples/testsets/abc120-d/out/sample_03.txt b/examples/testsets/abc120-d/out/sample_03.txt
@@ -0,0 +1 @@
+1

Original file line number	Diff line number	Diff line change
`@@ -12,5 +12,9 @@`
`12`	`12`	`name: "ABC049 / ARC065: C - 白昼夢 / Daydream",`
`13`	`13`	`matching: ExactWords,`
`14`	`14`	`),`
	`15`	`+ "abc120-d": (`
	`16`	`+ name: "ABC120: D - Decayed Bridges ",`
	`17`	`+ matching: ExactWords,`
	`18`	`+ ),`
`15`	`19`	`}`
`16`	`20`	`)`
-Original file line number
+Diff line change
@@ @@ -0,0 +1,6 @@ @@
 +4 5
 +1 2
 +3 4
 +1 3
 +2 3
 +1 4