Add an example for AGC023-A

qryxip · qryxip · commit 2ce36821f775 · 2020-05-19T00:20:07.000+09:00
diff --git a/examples/agc023-a.rs b/examples/agc023-a.rs
@@ -0,0 +1,35 @@
+// https://atcoder.jp/contests/agc023/tasks/agc023_a
+//
+// 以下のクレートを使用。
+//
+// - `itertools-num`
+// - `maplit`
+// - `proconio`
+
+use itertools_num::ItertoolsNum as _;
+use maplit::hashmap;
+use proconio::input;
+
+fn main() {
+    // `proconio::input!`。
+    //
+    // https://docs.rs/proconio/0.3.6/proconio/macro.input.html
+    input! {
+        r#as: [i64],
+    }
+
+    // `0`が一つが入ったcounterを`maplit::hashmap!`で作る。
+    //
+    // https://docs.rs/maplit/1/maplit/macro.hashmap.html
+    let mut counter = hashmap!(0 => 1u64); // `0`が一つが入ったcounterを`maplit::hashmap!`で作る。
+
+    // `<_ as itertools_num::ItertoolsNum>::cumsum`で作られた一次元累積和のイテレータを、`Vec`にせずにそのまま`for`文で回す。
+    //
+    // https://docs.rs/itertools-num/0.1/itertools_num/trait.ItertoolsNum.html#method.cumsum
+    for sum in r#as.into_iter().cumsum() {
+        *counter.entry(sum).or_insert(0) += 1;
+    }
+
+    let ans = counter.values().map(|v| v * (v - 1) / 2).sum::<u64>();
+    println!("{}", ans);
+}
diff --git a/test-examples.toml b/test-examples.toml
@@ -142,6 +142,13 @@ url = "https://atcoder.jp/contests/abc151/tasks/abc151_d"
 matching = "Words"
 meta = { using = ["itertools", "ndarray", "proconio", "smallvec"] }
 
+[examples.agc023-a]
+type = "Normal"
+name = "AGC023: A - Zero-Sum Ranges"
+url = "https://atcoder.jp/contests/agc023/tasks/agc023_a"
+matching = "Words"
+meta = { using = ["itertools-num", "maplit", "proconio"] }
+
 [examples.apg4b-a]
 type = "Normal"
 name = "APG4b: A - 1.00.はじめに"
