Add a test for APG4b-EX25

qryxip · qryxip · commit f4c96aa90862 · 2019-12-16T23:27:58.000+09:00
diff --git a/examples/apg4b-ex25.rs b/examples/apg4b-ex25.rs
@@ -0,0 +1,87 @@
+use fixedbitset::FixedBitSet;
+use itertools::Itertools as _;
+
+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 = read!(usize);
+    // let a = read!([usize; n]);
+    // let m = read!(usize);
+    // let b = read!([usize; m]);
+    // let arg0 = read!(String);
+    // let args = read!([usize; if arg0 == "subtract" { 1 } else { 0 }]);
+
+    use proconio::input;
+
+    input! {
+        n: usize,
+        a: [usize; n],
+        m: usize,
+        b: [usize; m],
+        arg0: String,
+        args: [usize; if arg0 == "subtract" { 1 } else { 0 }],
+    }
+
+    let (a, b) = (a.into_iter().collect(), b.into_iter().collect());
+
+    print_set(&match (&*arg0, &*args) {
+        ("intersection", []) => intersection(&a, &b),
+        ("union_set", []) => union_set(&a, &b),
+        ("symmetric_diff", []) => symmetric_diff(&a, &b),
+        ("subtract", &[x]) => subtract(a, x),
+        ("increment", []) => increment(&a),
+        ("decrement", []) => decrement(&a),
+        _ => unreachable!(),
+    });
+}
+
+fn print_set(set: &FixedBitSet) {
+    println!("{}", (0..50).filter(|&i| set[i]).format(" "));
+}
+
+fn intersection(a: &FixedBitSet, b: &FixedBitSet) -> FixedBitSet {
+    a & b
+}
+
+fn union_set(a: &FixedBitSet, b: &FixedBitSet) -> FixedBitSet {
+    a | b
+}
+
+fn symmetric_diff(a: &FixedBitSet, b: &FixedBitSet) -> FixedBitSet {
+    a ^ b
+}
+
+fn subtract(mut a: FixedBitSet, x: usize) -> FixedBitSet {
+    // > xは存在することが保証される。
+    a.set(x, false);
+    a
+}
+
+fn increment(a: &FixedBitSet) -> FixedBitSet {
+    a.ones().map(|x| (x + 1) % 50).collect()
+}
+
+fn decrement(a: &FixedBitSet) -> FixedBitSet {
+    a.ones().map(|x| (x + 49) % 50).collect()
+}
diff --git a/examples/tests.ron b/examples/tests.ron
@@ -7,6 +7,10 @@
             name: "APG4b: A - 1.00.はじめに",
             matching: ExactWhole,
         ),
+        "apg4b-ex25": (
+            name: "APG4b: EX25 - 集合の操作 / 3.05",
+            matching: ExactWords,
+        ),
         "practice-a": (
             name: "practice contest: A - Welcome to AtCoder",
             matching: ExactWords,
diff --git a/examples/testsets/apg4b-ex25/in/sample_01.txt b/examples/testsets/apg4b-ex25/in/sample_01.txt
@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+intersection
diff --git a/examples/testsets/apg4b-ex25/in/sample_02.txt b/examples/testsets/apg4b-ex25/in/sample_02.txt
@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+union_set
diff --git a/examples/testsets/apg4b-ex25/in/sample_03.txt b/examples/testsets/apg4b-ex25/in/sample_03.txt
@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+symmetric_diff
diff --git a/examples/testsets/apg4b-ex25/in/sample_04.txt b/examples/testsets/apg4b-ex25/in/sample_04.txt
@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+subtract 2
diff --git a/examples/testsets/apg4b-ex25/in/sample_05.txt b/examples/testsets/apg4b-ex25/in/sample_05.txt
@@ -0,0 +1,5 @@
+3
+0 1 49
+3
+1 2 3
+increment
diff --git a/examples/testsets/apg4b-ex25/in/sample_06.txt b/examples/testsets/apg4b-ex25/in/sample_06.txt
@@ -0,0 +1,5 @@
+3
+0 1 49
+3
+1 2 3
+decrement
diff --git a/examples/testsets/apg4b-ex25/out/sample_01.txt b/examples/testsets/apg4b-ex25/out/sample_01.txt
@@ -0,0 +1 @@
+1 2
diff --git a/examples/testsets/apg4b-ex25/out/sample_02.txt b/examples/testsets/apg4b-ex25/out/sample_02.txt
@@ -0,0 +1 @@
+0 1 2 3
diff --git a/examples/testsets/apg4b-ex25/out/sample_03.txt b/examples/testsets/apg4b-ex25/out/sample_03.txt
@@ -0,0 +1 @@
+0 3
diff --git a/examples/testsets/apg4b-ex25/out/sample_04.txt b/examples/testsets/apg4b-ex25/out/sample_04.txt
@@ -0,0 +1 @@
+0 1
diff --git a/examples/testsets/apg4b-ex25/out/sample_05.txt b/examples/testsets/apg4b-ex25/out/sample_05.txt
@@ -0,0 +1 @@
+0 1 2
diff --git a/examples/testsets/apg4b-ex25/out/sample_06.txt b/examples/testsets/apg4b-ex25/out/sample_06.txt
@@ -0,0 +1 @@
+0 48 49

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +3
 +0 1 2
 +3
 +1 2 3
 +intersection