Add an example for AGC020-C

qryxip · qryxip · commit 44be0d217eca · 2020-05-19T00:20:09.000+09:00
diff --git a/examples/agc020-c.rs b/examples/agc020-c.rs
@@ -0,0 +1,35 @@
+// https://atcoder.jp/contests/agc020/tasks/agc020_c
+//
+// 以下のクレートを使用。
+// - `bitset-fixed`
+// - `proconio`
+
+use bitset_fixed::BitSet;
+use proconio::input;
+
+fn main() {
+    // `proconio::input!`。
+    //
+    // https://docs.rs/proconio/0.3.6/proconio/macro.input.html
+    input! {
+        r#as: [usize],
+    }
+
+    // `bitset-fixed`のREADMEにある解法。
+
+    let sum = r#as.iter().sum::<usize>();
+
+    // サイズ`sum + 1`のbit setに`fixed_bitset::BitSet`を使う。
+    // これは`fixedbitset::FixedBitSet`と違いビットシフトが可能。
+    // (`bitvec`クレートだと可能。提案しておけばよかったか..)
+    //
+    // https://docs.rs/bitset-fixed/0.1/bitset_fixed/struct.BitSet.html
+    let mut dp = BitSet::new(sum + 1);
+    dp.set(0, true);
+    for a in r#as {
+        dp |= &(&dp << a);
+    }
+
+    let ans = ((sum + 1) / 2..).find(|&i| dp[i]).unwrap();
+    println!("{}", ans);
+}
diff --git a/test-examples.toml b/test-examples.toml
@@ -226,6 +226,13 @@ url = "https://atcoder.jp/contests/abc160/tasks/abc160_e"
 matching = "Words"
 meta = { using = ["itertools", "proconio"] }
 
+[examples.agc020-c]
+type = "Normal"
+name = "AGC020: C - Median Sum"
+url = "https://atcoder.jp/contests/agc020/tasks/agc020_c"
+matching = "Words"
+meta = { using = ["bitset-fixed", "proconio"] }
+
 [examples.agc023-a]
 type = "Normal"
 name = "AGC023: A - Zero-Sum Ranges"
