Add an example for ABC160-E

qryxip · qryxip · commit 97f555d96239 · 2020-05-19T00:20:09.000+09:00
diff --git a/examples/abc160-e.rs b/examples/abc160-e.rs
@@ -0,0 +1,40 @@
+// https://atcoder.jp/contests/abc160/tasks/abc160_e
+//
+// 以下のクレートを使用。
+// - `itertools`
+// - `proconio`
+
+use itertools::Itertools as _;
+use proconio::input;
+
+use std::cmp::{self, Reverse};
+
+fn main() {
+    // `proconio::input!`はオリジナルの`input!`とは違い、`mut $ident`の形式で入力を読むことができる。
+    //
+    // https://docs.rs/proconio/0.3.6/proconio/macro.input.html
+    input! {
+        x: usize,
+        y: usize,
+        a: usize,
+        b: usize,
+        c: usize,
+        mut ps: [u64; a],
+        mut qs: [u64; b],
+        mut rs: [u64; c],
+    }
+
+    ps.sort_unstable_by_key(|&p| Reverse(p));
+    qs.sort_unstable_by_key(|&q| Reverse(q));
+    rs.sort_unstable_by_key(|&r| Reverse(r));
+
+    // `itertools::Itertools::kmerge_by`で降順のままでマージする。
+    //
+    // https://docs.rs/itertools/0.9/itertools/trait.Itertools.html#method.kmerge_by
+    let ans = vec![&ps[..x], &qs[..y], &rs[..cmp::min(x + y, c)]]
+        .into_iter()
+        .kmerge_by(|v1, v2| v1 > v2)
+        .take(x + y)
+        .sum::<u64>();
+    println!("{}", ans);
+}
diff --git a/test-examples.toml b/test-examples.toml
@@ -219,6 +219,13 @@ url = "https://atcoder.jp/contests/abc157/tasks/abc157_e"
 matching = "Words"
 meta = { using = ["alga", "proconio", "whiteread"] }
 
+[examples.abc160-e]
+type = "Normal"
+name = "ABC160: E - Red and Green Apples"
+url = "https://atcoder.jp/contests/abc160/tasks/abc160_e"
+matching = "Words"
+meta = { using = ["itertools", "proconio"] }
+
 [examples.agc023-a]
 type = "Normal"
 name = "AGC023: A - Zero-Sum Ranges"
