Add an example for ABC166-B

qryxip · qryxip · commit c867fecdc848 · 2020-05-19T00:20:10.000+09:00
diff --git a/examples/abc166-b.rs b/examples/abc166-b.rs
@@ -0,0 +1,38 @@
+// https://atcoder.jp/contests/abc166/tasks/abc166_b
+//
+// 以下のクレートを使用。
+// - `itertools`
+// - `proconio`
+
+use itertools::Itertools as _;
+use proconio::{input, marker::Usize1};
+
+fn main() {
+    // `proconio::input!`には`n: usize, xs: [T; n]`のかわりに`xs: [T]`と書ける機能がある。
+    // この機能を用いることで今回のような入力をシンプルにできる。
+    //
+    // https://docs.rs/proconio/0.3.6/proconio/macro.input.html
+    input! {
+        n: usize,
+        ass: [[Usize1]],
+    }
+
+    // `Itertools::unique`で要素の重複が省かれたイテレータが手に入る。
+    //
+    // https://docs.rs/itertools/0.9.0/itertools/trait.Itertools.html#method.unique
+    let ans = n - ass.into_iter().flatten().unique().count();
+
+    // あるいは`fixedbitset::FixedBitSet`を使っても良い。
+    // `FixedBitSet`は`FromIterator<usize>`であり、`count_ones`という「`1`」を数えるメソッドを持つ。
+    //
+    // https://docs.rs/fixedbitset/0.2.0/fixedbitset/struct.FixedBitSet.html
+    // https://docs.rs/fixedbitset/0.2.0/fixedbitset/struct.FixedBitSet.html#method.count_ones
+    //use fixedbitset::FixedBitSet;
+    //let ans = n - ass
+    //    .into_iter()
+    //    .flatten()
+    //    .collect::<FixedBitSet>()
+    //    .count_ones(..);
+
+    println!("{}", ans);
+}
diff --git a/test-examples.toml b/test-examples.toml
@@ -247,6 +247,13 @@ url = "https://atcoder.jp/contests/abc165/tasks/abc165_c"
 matching = "Words"
 meta = { using = ["itertools", "proconio"] }
 
+[examples.abc166-b]
+type = "Normal"
+name = "ABC166 - B - Trick or Treat"
+url = "https://atcoder.jp/contests/abc166/tasks/abc166_b"
+matching = "Words"
+meta = { using = ["itertools", "proconio"] }
+
 [examples.agc020-c]
 type = "Normal"
 name = "AGC020: C - Median Sum"
