More tests (or rather, examples)

.cargo/config

@@ -1,3 +1,3 @@
 [alias]
 dep-tests = ["run", "--manifest-path", "./tools/dep-tests/Cargo.toml", "--"]
-test-with-generated-opts = ["run", "--manifest-path", "./tools/test-with-generated-opts/Cargo.toml", "--"]
+test-examples = ["run", "--manifest-path", "./tools/test-examples/Cargo.toml", "--"]

.github/workflows/ci.yml

@@ -43,20 +43,25 @@ jobs:
           command: fmt
           args: --manifest-path ./tools/dep-tests/Cargo.toml -- --check
 
-      - name: '`cargo fmt --manifest-path ./tools/test-with-generated-opts/Cargo.toml -- --check`'
+      - name: '`cargo fmt --manifest-path ./tools/test-examples/Cargo.toml -- --check`'
         uses: actions-rs/cargo@v1
         with:
           command: fmt
-          args: --manifest-path ./tools/test-with-generated-opts/Cargo.toml -- --check
+          args: --manifest-path ./tools/test-examples/Cargo.toml -- --check
 
-  test-with-generated-opts:
-    name: test-with-generated-opts
+  test-examples:
+    name: test-examples
     runs-on: ubuntu-18.04
 
     steps:
       - name: Checkout
         uses: actions/checkout@v1
 
+      - name: setup-python
+        uses: actions/setup-python@v1
+        with:
+          python-version: '3.8'
+
       - name: rust-toolchain
         uses: actions-rs/toolchain@v1
         with:
@@ -70,25 +75,22 @@ jobs:
           command: install
           args: --git https://github.com/rust-lang-ja/atcoder-rustc-dep-option-generator
 
-      - name: '`cargo clippy --all-features --manifest-path ./tools/test-with-generated-opts/Cargo.toml -- -D warnings`'
+      - name: '`cargo clippy --all-features --manifest-path ./tools/test-examples/Cargo.toml -- -D warnings`'
         uses: actions-rs/cargo@v1
         with:
           command: clippy
-          args: --all-features --manifest-path ./tools/test-with-generated-opts/Cargo.toml -- -D warnings
+          args: --all-features --manifest-path ./tools/test-examples/Cargo.toml -- -D warnings
 
       - name: '`cargo build --all-features --release`'
         uses: actions-rs/cargo@v1
         with:
           command: build
           args: --all-features --release
 
-      - name: '`chmod -R a=rX-w ./target`'
-        run: chmod -R a=rX-w ./target
-
-      - name: '`cargo test-with-generated-opts`'
+      - name: '`cargo test-examples`'
         uses: actions-rs/cargo@v1
         with:
-          command: test-with-generated-opts
+          command: test-examples
 
   build:
     strategy:
@@ -176,17 +178,17 @@ jobs:
           override: true
           profile: default
 
-      - name: '`cargo clippy ${{ matrix.features }} -- -D warnings`'
+      - name: '`cargo clippy --all-targets --profile test ${{ matrix.features }} -- -D warnings`'
         uses: actions-rs/cargo@v1
         with:
           command: clippy
-          args: ${{ matrix.features }} -- -D warnings
+          args: --all-targets --profile test ${{ matrix.features }} -- -D warnings
 
-      - name: '`cargo test ${{ matrix.features }} --no-fail-fast`'
+      - name: '`cargo test --all-targets ${{ matrix.features }} --no-fail-fast`'
         uses: actions-rs/cargo@v1
         with:
           command: test
-          args: ${{ matrix.features }} --no-fail-fast
+          args: --all-targets ${{ matrix.features }} --no-fail-fast
 
       - name: '`cargo run --release`'
         uses: actions-rs/cargo@v1

.gitignore

@@ -1,5 +1,5 @@
 /tools/dep-tests/Cargo.lock
-/tools/test-with-generated-opts/Cargo.lock
+/tools/test-examples/Cargo.lock
 **/target/
 **/*.rs.bk
 **/*~

examples/abc054-c.rs

@@ -1,55 +1,21 @@
 // https://atcoder.jp/contests/abc054/tasks/abc054_c
 
-use petgraph::csr::Csr;
-use petgraph::Undirected;
+use itertools::Itertools as _;
+use petgraph::graph::UnGraph;
+use proconio::input;
+use proconio::marker::Usize1;
 
 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, m) = read!((usize, usize));
-    // let mut abs = read!([(_1based, _1based); m]);
-
-    use proconio::input;
-    use proconio::marker::Usize1;
-
     input! {
         n: usize,
-        m: usize,
-        mut abs: [(Usize1, Usize1); m],
+        abs: [(Usize1, Usize1)],
     }
 
-    abs.sort();
-    let mut g = Csr::<(), (), Undirected, usize>::with_nodes(n);
-    for (a, b) in abs {
-        g.add_edge(a, b, ());
-    }
-    let mut ans = 0;
-    let mut es = (0..n).collect::<Vec<_>>();
-    permutohedron::heap_recursive(&mut es, |es| {
-        if es[0] == 0 && es.windows(2).all(|w| g.contains_edge(w[0], w[1])) {
-            ans += 1;
-        }
-    });
+    let graph = UnGraph::<(), (), usize>::from_edges(abs);
+    let ans = graph
+        .node_indices()
+        .permutations(n)
+        .filter(|p| p[0].index() == 0 && p.windows(2).all(|w| graph.contains_edge(w[0], w[1])))
+        .count();
     println!("{}", ans);
 }

examples/abc057-b-naive.rs

@@ -0,0 +1,32 @@
+// https://atcoder.jp/contests/abc057/tasks/abc057_b
+
+use std::io::{self, Read};
+
+fn main() {
+    let mut input = read_to_static(io::stdin()).split_whitespace();
+    macro_rules! read {
+        ([$tt:tt]) => (read!([$tt; read!(usize)]));
+        ([$tt:tt; $n:expr]) => ((0..$n).map(|_| read!($tt)).collect::<Vec<_>>());
+        (($($tt:tt),+)) => (($(read!($tt)),*));
+        ($ty:ty) => (input.next().unwrap().parse::<$ty>().unwrap());
+    }
+
+    let (n, m) = read!((usize, usize));
+    let (abs, cds) = read!(([(i64, i64); n], [(i64, i64); m]));
+
+    for (a, b) in abs {
+        let j = (0..m)
+            .min_by_key(|&j| {
+                let (c, d) = cds[j];
+                (a - c).abs() + (b - d).abs()
+            })
+            .unwrap();
+        println!("{}", j + 1);
+    }
+}
+
+fn read_to_static(mut source: impl Read) -> &'static str {
+    let mut input = "".to_owned();
+    source.read_to_string(&mut input).unwrap();
+    Box::leak(input.into_boxed_str())
+}

examples/abc057-b-proconio.rs

@@ -0,0 +1,22 @@
+// https://atcoder.jp/contests/abc057/tasks/abc057_b
+
+use proconio::input;
+
+fn main() {
+    input! {
+        n: usize,
+        m: usize,
+        abs: [(i64, i64); n],
+        cds: [(i64, i64); m],
+    }
+
+    for (a, b) in abs {
+        let j = (0..m)
+            .min_by_key(|&j| {
+                let (c, d) = cds[j];
+                (a - c).abs() + (b - d).abs()
+            })
+            .unwrap();
+        println!("{}", j + 1);
+    }
+}

examples/abc057-b-text-io.rs

@@ -0,0 +1,26 @@
+// https://atcoder.jp/contests/abc057/tasks/abc057_b
+
+#![allow(clippy::many_single_char_names, clippy::try_err)]
+
+use text_io::{read, try_read, try_scan};
+
+fn main() {
+    let n: usize = read!();
+    let m: usize = read!();
+    let abs = (0..n)
+        .map(|_| (read!(), read!()))
+        .collect::<Vec<(i64, i64)>>();
+    let cds = (0..m)
+        .map(|_| (read!(), read!()))
+        .collect::<Vec<(i64, i64)>>();
+
+    for (a, b) in abs {
+        let j = (0..m)
+            .min_by_key(|&j| {
+                let (c, d) = cds[j];
+                (a - c).abs() + (b - d).abs()
+            })
+            .unwrap();
+        println!("{}", j + 1);
+    }
+}

examples/abc057-b-whiteread.rs

@@ -0,0 +1,21 @@
+// https://atcoder.jp/contests/abc057/tasks/abc057_b
+
+use whiteread::Reader;
+
+fn main() {
+    let mut rdr = Reader::from_stdin_naive();
+
+    let (n, m) = rdr.p::<(usize, usize)>();
+    let abs = (0..n).map(|_| rdr.p()).collect::<Vec<(i64, i64)>>();
+    let cds = (0..m).map(|_| rdr.p()).collect::<Vec<(i64, i64)>>();
+
+    for (a, b) in abs {
+        let j = (0..m)
+            .min_by_key(|&j| {
+                let (c, d) = cds[j];
+                (a - c).abs() + (b - d).abs()
+            })
+            .unwrap();
+        println!("{}", j + 1);
+    }
+}

examples/abc073-d.rs

@@ -0,0 +1,50 @@
+// https://atcoder.jp/contests/abc073/tasks/abc073_d
+
+use itertools::Itertools as _;
+use num::traits::One;
+use petgraph::graph::{IndexType, NodeIndex, UnGraph};
+use proconio::input;
+use proconio::source::{Readable, Source};
+
+use std::collections::HashMap;
+use std::io::BufRead;
+use std::marker::PhantomData;
+use std::ops::Sub;
+
+fn main() {
+    input! {
+        _n: usize,
+        m: usize,
+        r: usize,
+        rs: [NodeIndex1<u32>; r],
+        abcs: [(NodeIndex1<u32>, NodeIndex1<u32>, u32); m],
+    }
+
+    let graph = UnGraph::<(), u32>::from_edges(abcs);
+
+    let dijkstra = rs
+        .iter()
+        .map(|&r| {
+            let dijkstra = petgraph::algo::dijkstra(&graph, r, None, |e| *e.weight());
+            (r, dijkstra)
+        })
+        .collect::<HashMap<_, _>>();
+
+    let ans = rs
+        .into_iter()
+        .permutations(r)
+        .map(|rs| rs.windows(2).map(|w| dijkstra[&w[0]][&w[1]]).sum::<u32>())
+        .min()
+        .unwrap();
+    println!("{}", ans);
+}
+
+struct NodeIndex1<Ix>(PhantomData<fn() -> Ix>);
+
+impl<Ix: IndexType + Readable<Output = Ix> + One + Sub<Output = Ix>> Readable for NodeIndex1<Ix> {
+    type Output = NodeIndex<Ix>;
+
+    fn read<R: BufRead, S: Source<R>>(source: &mut S) -> NodeIndex<Ix> {
+        NodeIndex::from(Ix::read(source) - Ix::one())
+    }
+}