Solve A in agc039

Author: k-yomo

atcoder/rust/agc039/Cargo.lock

@@ -0,0 +1,80 @@
+[package]
+name = "agc039"
+version = "0.1.0"
+authors = ["k-yomo <kanji.yy@gmail.com>"]
+edition = "2018"
+
+[package.metadata.cargo-compete]
+config = "../compete.toml"
+
+[package.metadata.cargo-compete.bin]
+a = { name = "agc039-a", problem = { platform = "atcoder", contest = "agc039", index = "A", url = "https://atcoder.jp/contests/agc039/tasks/agc039_a" } }
+b = { name = "agc039-b", problem = { platform = "atcoder", contest = "agc039", index = "B", url = "https://atcoder.jp/contests/agc039/tasks/agc039_b" } }
+c = { name = "agc039-c", problem = { platform = "atcoder", contest = "agc039", index = "C", url = "https://atcoder.jp/contests/agc039/tasks/agc039_c" } }
+d = { name = "agc039-d", problem = { platform = "atcoder", contest = "agc039", index = "D", url = "https://atcoder.jp/contests/agc039/tasks/agc039_d" } }
+e = { name = "agc039-e", problem = { platform = "atcoder", contest = "agc039", index = "E", url = "https://atcoder.jp/contests/agc039/tasks/agc039_e" } }
+f = { name = "agc039-f", problem = { platform = "atcoder", contest = "agc039", index = "F", url = "https://atcoder.jp/contests/agc039/tasks/agc039_f" } }
+
+[[bin]]
+name = "agc039-a"
+path = "src/bin/a.rs"
+
+[[bin]]
+name = "agc039-b"
+path = "src/bin/b.rs"
+
+[[bin]]
+name = "agc039-c"
+path = "src/bin/c.rs"
+
+[[bin]]
+name = "agc039-d"
+path = "src/bin/d.rs"
+
+[[bin]]
+name = "agc039-e"
+path = "src/bin/e.rs"
+
+[[bin]]
+name = "agc039-f"
+path = "src/bin/f.rs"
+[dependencies]
+num = "=0.2.1"
+num-bigint = "=0.2.6"
+num-complex = "=0.2.4"
+num-integer = "=0.1.42"
+num-iter = "=0.1.40"
+num-rational = "=0.2.4"
+num-traits = "=0.2.11"
+num-derive = "=0.3.0"
+ndarray = "=0.13.0"
+nalgebra = "=0.20.0"
+alga = "=0.9.3"
+libm = "=0.2.1"
+rand = { version = "=0.7.3", features = ["small_rng"] }
+getrandom = "=0.1.14"
+rand_chacha = "=0.2.2"
+rand_core = "=0.5.1"
+rand_hc = "=0.2.0"
+rand_pcg = "=0.2.1"
+rand_distr = "=0.2.2"
+petgraph = "=0.5.0"
+indexmap = "=1.3.2"
+regex = "=1.3.6"
+lazy_static = "=1.4.0"
+ordered-float = "=1.0.2"
+ascii = "=1.0.0"
+permutohedron = "=0.2.4"
+superslice = "=1.0.0"
+itertools = "=0.9.0"
+itertools-num = "=0.1.3"
+maplit = "=1.0.2"
+either = "=1.5.3"
+im-rc = "=14.3.0"
+fixedbitset = "=0.2.0"
+bitset-fixed = "=0.1.0"
+proconio = { version = "=0.3.6", features = ["derive"] }
+text_io = "=0.1.8"
+whiteread = "=0.5.0"
+rustc-hash = "=1.1.0"
+smallvec = "=1.2.0"

atcoder/rust/agc039/Cargo.toml

@@ -0,0 +1,61 @@
+#![allow(unused_imports)]
+
+use std::cmp::*;
+use std::collections::*;
+use std::io::Write;
+use std::ops::Bound::*;
+
+use itertools::__std_iter::once;
+use itertools::*;
+use itertools_num::ItertoolsNum;
+use proconio::marker::*;
+use proconio::*;
+use superslice::*;
+
+fn main() {
+    input! {
+        s: Chars,
+        k: usize,
+    }
+
+    if s.len() == 1 {
+        return println!("{}", k / 2);
+    }
+
+    let dup_counts = s
+        .iter()
+        .map(|c| (c, 1))
+        .coalesce(|(prev_c, prev_dup), (cur_c, cur_dup)| {
+            if prev_c == cur_c {
+                Ok((prev_c, prev_dup + cur_dup))
+            } else {
+                Err(((prev_c, prev_dup), (cur_c, cur_dup)))
+            }
+        })
+        .map(|(_, count)| count)
+        .filter(|&count| count > 1)
+        .collect::<Vec<usize>>();
+    if dup_counts.len() == 1 {
+        return println!("{}", s.len() * k / 2);
+    } else {
+        if s[0] == s[s.len() - 1] {
+            let len = dup_counts.len();
+            let mut dup_count = 0;
+            dup_count += dup_counts[0] / 2;
+            dup_count += dup_counts[len - 1] / 2;
+            dup_count += (dup_counts[0] + dup_counts[len - 1]) / 2 * (k - 1);
+            if len >= 3 {
+                dup_count += dup_counts[1..(len - 1)]
+                    .iter()
+                    .map(|count| count / 2 * k)
+                    .sum::<usize>();
+            }
+            println!("{}", dup_count)
+        } else {
+            println!(
+                "{}",
+                dup_counts.iter().map(|count| count / 2 * k).sum::<usize>()
+            )
+        }
+    }
+}

atcoder/rust/agc039/src/bin/a.rs

@@ -0,0 +1,19 @@
+#![allow(unused_imports)]
+
+use std::cmp::*;
+use std::collections::*;
+use std::io::Write;
+use std::ops::Bound::*;
+
+use itertools::*;
+use itertools::__std_iter::once;
+use itertools_num::ItertoolsNum;
+use proconio::*;
+use proconio::marker::*;
+use superslice::*;
+
+fn main() {
+    input! {
+
+    }
+}

atcoder/rust/agc039/src/bin/b.rs

@@ -0,0 +1,19 @@
+#![allow(unused_imports)]
+
+use std::cmp::*;
+use std::collections::*;
+use std::io::Write;
+use std::ops::Bound::*;
+
+use itertools::*;
+use itertools::__std_iter::once;
+use itertools_num::ItertoolsNum;
+use proconio::*;
+use proconio::marker::*;
+use superslice::*;
+
+fn main() {
+    input! {
+
+    }
+}

atcoder/rust/agc039/src/bin/c.rs

@@ -0,0 +1,19 @@
+#![allow(unused_imports)]
+
+use std::cmp::*;
+use std::collections::*;
+use std::io::Write;
+use std::ops::Bound::*;
+
+use itertools::*;
+use itertools::__std_iter::once;
+use itertools_num::ItertoolsNum;
+use proconio::*;
+use proconio::marker::*;
+use superslice::*;
+
+fn main() {
+    input! {
+
+    }
+}

atcoder/rust/agc039/src/bin/d.rs

@@ -0,0 +1,19 @@
+#![allow(unused_imports)]
+
+use std::cmp::*;
+use std::collections::*;
+use std::io::Write;
+use std::ops::Bound::*;
+
+use itertools::*;
+use itertools::__std_iter::once;
+use itertools_num::ItertoolsNum;
+use proconio::*;
+use proconio::marker::*;
+use superslice::*;
+
+fn main() {
+    input! {
+
+    }
+}

atcoder/rust/agc039/src/bin/e.rs

@@ -0,0 +1,19 @@
+#![allow(unused_imports)]
+
+use std::cmp::*;
+use std::collections::*;
+use std::io::Write;
+use std::ops::Bound::*;
+
+use itertools::*;
+use itertools::__std_iter::once;
+use itertools_num::ItertoolsNum;
+use proconio::*;
+use proconio::marker::*;
+use superslice::*;
+
+fn main() {
+    input! {
+
+    }
+}

atcoder/rust/agc039/src/bin/f.rs

@@ -0,0 +1,26 @@
+---
+type: Batch
+timelimit: 2s
+match: Lines
+
+cases:
+  - name: sample1
+    in: |
+      issii
+      2
+    out: |
+      4
+  - name: sample2
+    in: |
+      qq
+      81
+    out: |
+      81
+  - name: sample3
+    in: |
+      cooooooooonteeeeeeeeeest
+      999993333
+    out: |
+      8999939997
+
+extend: []

atcoder/rust/agc039/testcases/a.yml

@@ -0,0 +1,34 @@
+---
+type: Batch
+timelimit: 2s
+match: Lines
+
+cases:
+  - name: sample1
+    in: |
+      2
+      01
+      10
+    out: |
+      2
+  - name: sample2
+    in: |
+      3
+      011
+      101
+      110
+    out: |
+      -1
+  - name: sample3
+    in: |
+      6
+      010110
+      101001
+      010100
+      101000
+      100000
+      010000
+    out: |
+      4
+
+extend: []