Add a test for APG4b-EX25

Author: qryxip

examples/apg4b-ex25.rs

@@ -0,0 +1,89 @@
+// https://atcoder.jp/contests/APG4b/tasks/APG4b_bx
+
+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()
+}

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,

examples/testsets/apg4b-ex25/in/sample_01.txt

@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+intersection

examples/testsets/apg4b-ex25/in/sample_02.txt

@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+union_set

examples/testsets/apg4b-ex25/in/sample_03.txt

@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+symmetric_diff

examples/testsets/apg4b-ex25/in/sample_04.txt

@@ -0,0 +1,5 @@
+3
+0 1 2
+3
+1 2 3
+subtract 2

examples/testsets/apg4b-ex25/in/sample_05.txt

@@ -0,0 +1,5 @@
+3
+0 1 49
+3
+1 2 3
+increment

examples/testsets/apg4b-ex25/in/sample_06.txt

@@ -0,0 +1,5 @@
+3
+0 1 49
+3
+1 2 3
+decrement

examples/testsets/apg4b-ex25/out/sample_01.txt

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

examples/testsets/apg4b-ex25/out/sample_02.txt

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

examples/testsets/apg4b-ex25/out/sample_03.txt

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

examples/testsets/apg4b-ex25/out/sample_04.txt

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

examples/testsets/apg4b-ex25/out/sample_05.txt

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

examples/testsets/apg4b-ex25/out/sample_06.txt

@@ -0,0 +1 @@
+0 48 49