Add a test for ABC129-F

Author: qryxip

examples/abc129-f.rs

@@ -0,0 +1,174 @@
+// https://atcoder.jp/contests/abc129/tasks/abc129_f
+
+use derive_more::{Display, FromStr};
+use ndarray::{array, Array2, LinalgScalar};
+use num::integer::ExtendedGcd;
+use num::{CheckedDiv, Integer as _, One, PrimInt, Unsigned, Zero};
+
+use std::cell::Cell;
+use std::cmp;
+use std::convert::{TryFrom as _, TryInto as _};
+use std::ops::{Add, Div, Mul, Sub};
+
+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 (l, a, b, m) = read!((u64, u64, u64, u64));
+
+    use proconio::input;
+
+    input! {
+        l: u64,
+        a: u64,
+        b: u64,
+        m: u64,
+    }
+
+    MOD.with(|cell| cell.set(m));
+
+    let count = |d: u32| -> _ {
+        let count = |sup: u64| cmp::min((cmp::max(sup, a + 1) - a - 1) / b + u64::from(b < sup), l);
+        count(10u64.pow(d)) - count(10u64.pow(d) / 10)
+    };
+
+    let ans = (1..=18).fold(array![[Z::new(0), Z::new(a), Z::new(1)]], |acc, d| {
+        acc.dot(
+            &array![
+                [Z::new(10u64.pow(d)), Z::new(0), Z::new(0)],
+                [Z::new(1), Z::new(1), Z::new(0)],
+                [Z::new(0), Z::new(b), Z::new(1)],
+            ]
+            .matrix_power(count(d)),
+        )
+    })[(0, 0)];
+    println!("{}", ans);
+}
+
+trait Array2Ext {
+    fn matrix_power<E: PrimInt + Unsigned>(&self, exp: E) -> Self;
+}
+
+impl<S: LinalgScalar> Array2Ext for Array2<S> {
+    fn matrix_power<E: PrimInt + Unsigned>(&self, exp: E) -> Self {
+        let (mut base, mut exp, mut acc) = (self.clone(), exp, Self::eye(self.nrows()));
+        while exp > E::zero() {
+            if (exp & E::one()) == E::one() {
+                acc = acc.dot(&base);
+            }
+            exp = exp / (E::one() + E::one());
+            base = base.dot(&base);
+        }
+        acc
+    }
+}
+
+thread_local! {
+    static MOD: Cell<u64> = Cell::new(0);
+}
+
+#[derive(FromStr, Display, Debug, Clone, Copy)]
+struct Z {
+    repr: u64,
+}
+
+impl Z {
+    fn new(mut val: u64) -> Self {
+        let modulus = MOD.with(Cell::get);
+        if val >= modulus {
+            val %= modulus;
+        }
+        Self { repr: val }
+    }
+}
+
+impl Zero for Z {
+    fn zero() -> Self {
+        Self { repr: 0 }
+    }
+
+    fn is_zero(&self) -> bool {
+        self.repr == 0
+    }
+}
+
+impl One for Z {
+    fn one() -> Self {
+        Self { repr: 1 }
+    }
+}
+
+impl Add for Z {
+    type Output = Self;
+
+    fn add(self, rhs: Self) -> Self {
+        Self::new(self.repr + rhs.repr)
+    }
+}
+
+impl Sub for Z {
+    type Output = Self;
+
+    fn sub(self, rhs: Self) -> Self {
+        debug_assert!(false, "really necessary?");
+
+        let repr = if self.repr < rhs.repr {
+            MOD.with(Cell::get) + self.repr - rhs.repr
+        } else {
+            self.repr - rhs.repr
+        };
+        Self { repr }
+    }
+}
+
+impl Mul for Z {
+    type Output = Self;
+
+    fn mul(self, rhs: Self) -> Self {
+        Self::new(self.repr * rhs.repr)
+    }
+}
+
+impl Div for Z {
+    type Output = Self;
+
+    fn div(self, rhs: Self) -> Self {
+        debug_assert!(false, "really necessary?");
+
+        self.checked_div(&rhs)
+            .unwrap_or_else(|| panic!("could not divide {} by {}", self, rhs))
+    }
+}
+
+impl CheckedDiv for Z {
+    fn checked_div(&self, rhs: &Self) -> Option<Self> {
+        let lhs = i64::try_from(self.repr).unwrap();
+        let rhs = i64::try_from(rhs.repr).unwrap();
+        let modulus = i64::try_from(MOD.with(Cell::get)).unwrap();
+        let ExtendedGcd { gcd, x, .. } = rhs.extended_gcd(&modulus);
+        match lhs % gcd {
+            0 => Some(Self::new((lhs * (x + modulus)).try_into().unwrap())),
+            _ => None,
+        }
+    }
+}

examples/tests.ron

@@ -43,6 +43,10 @@
             name: "ABC120: D - Decayed Bridges",
             matching: ExactWords,
         ),
+        "abc129-f": (
+            name: "ABC129: F - Takahashi's Basics in Education and Learning",
+            matching: ExactWords,
+        ),
         "abc142-d": (
             name: "ABC124: D - Disjoint Set of Common Divisors",
             matching: ExactWords,

examples/testsets/abc129-f/in/sample_01.txt

@@ -0,0 +1 @@
+5 3 4 10007

examples/testsets/abc129-f/in/sample_02.txt

@@ -0,0 +1 @@
+4 8 1 1000000

examples/testsets/abc129-f/in/sample_03.txt

@@ -0,0 +1 @@
+107 10000000000007 1000000000000007 998244353

examples/testsets/abc129-f/in/sub1_killer_01.txt

@@ -0,0 +1 @@
+2500000000000 4 4 998244353

examples/testsets/abc129-f/in/sub1_killer_02.txt

@@ -0,0 +1 @@
+99982444353 100 4 998244353

examples/testsets/abc129-f/in/sub1_killer_03.txt

@@ -0,0 +1 @@
+333333333333333333 3 3 998244353

examples/testsets/abc129-f/in/sub1_killer_04.txt

@@ -0,0 +1 @@
+1 999999999999999999 999999999999999999 998244353

examples/testsets/abc129-f/in/sub1_killer_05.txt

@@ -0,0 +1 @@
+998244353 629149506971940504 319836804 14876851

examples/testsets/abc129-f/in/sub1_killer_06.txt

@@ -0,0 +1 @@
+999918203 4248 723308865 998244353

examples/testsets/abc129-f/in/sub1_killer_07.txt

@@ -0,0 +1 @@
+100000000000 6 231 25200

examples/testsets/abc129-f/in/sub1_large_01.txt

@@ -0,0 +1 @@
+999999999999999999 1 1 1000000000

examples/testsets/abc129-f/in/sub1_large_02.txt

@@ -0,0 +1 @@
+11837588 221218607 84473754642 998244353

examples/testsets/abc129-f/in/sub1_large_03.txt

@@ -0,0 +1 @@
+81074056293 7047569542 12261087 29999997

examples/testsets/abc129-f/in/sub1_large_04.txt

@@ -0,0 +1 @@
+999999999999999999 1 1 998244353

examples/testsets/abc129-f/in/sub1_large_05.txt

@@ -0,0 +1 @@
+328613923524538 40980 3031 999999999

examples/testsets/abc129-f/in/sub1_large_06.txt

@@ -0,0 +1 @@
+10914883567412 1570 62840 99999999

examples/testsets/abc129-f/in/sub1_large_07.txt

@@ -0,0 +1 @@
+1448057007031098 11 690 881005587

examples/testsets/abc129-f/in/sub1_large_08.txt

@@ -0,0 +1 @@
+74358445905354 29310 13448 147413419

examples/testsets/abc129-f/in/sub1_large_09.txt

@@ -0,0 +1 @@
+120900801167206 994 8061 387420489

examples/testsets/abc129-f/in/sub1_rand_01.txt

@@ -0,0 +1 @@
+1380800559 8 721124722 998244353

examples/testsets/abc129-f/in/sub1_rand_02.txt

@@ -0,0 +1 @@
+105301221317 8 9457145 999999999

examples/testsets/abc129-f/in/sub1_rand_03.txt

@@ -0,0 +1 @@
+53562851 12672052437647 18553359394 99999999

examples/testsets/abc129-f/in/sub1_rand_04.txt

@@ -0,0 +1 @@
+7343 44375369427231 136194167038156 841293953

examples/testsets/abc129-f/in/sub1_rand_05.txt

@@ -0,0 +1 @@
+234481 8715183188079 4262710627175 9999999

examples/testsets/abc129-f/in/sub1_rand_06.txt

@@ -0,0 +1 @@
+335 2647488178484 2976541338296614 999918169

examples/testsets/abc129-f/in/sub1_rand_07.txt

@@ -0,0 +1 @@
+2508131942730888 5 397 387420489

examples/testsets/abc129-f/in/sub1_rand_08.txt

@@ -0,0 +1 @@
+151407055 44693807224027 6603923461 1000000000

examples/testsets/abc129-f/in/sub1_rand_09.txt

@@ -0,0 +1 @@
+17544042820 3 56987735 387420489

examples/testsets/abc129-f/in/sub1_rand_10.txt

@@ -0,0 +1 @@
+355391 6022223566881 25138147452 999958020

examples/testsets/abc129-f/in/sub1_small_01.txt

@@ -0,0 +1 @@
+18 912087068546306627 3164574410046630 996674566

examples/testsets/abc129-f/in/sub1_small_02.txt

@@ -0,0 +1 @@
+31940 139378099479290043 26943708742238 1000000000

examples/testsets/abc129-f/in/sub1_small_03.txt

@@ -0,0 +1 @@
+345 493299812499659259 1418510335032035 998244353

examples/testsets/abc129-f/in/sub1_small_04.txt

@@ -0,0 +1 @@
+3776 494095262613235502 19199433662734 999999

examples/testsets/abc129-f/in/sub1_small_05.txt

@@ -0,0 +1 @@
+461491 67429808854595636 2017247872269 276712227

examples/testsets/abc129-f/in/sub1_small_06.txt

@@ -0,0 +1 @@
+147 4222724580641548 6588820255446743 982883281

examples/testsets/abc129-f/out/sample_01.txt

@@ -0,0 +1 @@
+5563

examples/testsets/abc129-f/out/sample_02.txt

@@ -0,0 +1 @@
+891011

examples/testsets/abc129-f/out/sample_03.txt

@@ -0,0 +1 @@
+39122908

examples/testsets/abc129-f/out/sub1_killer_01.txt

@@ -0,0 +1 @@
+900161126

examples/testsets/abc129-f/out/sub1_killer_02.txt

@@ -0,0 +1 @@
+658481246

examples/testsets/abc129-f/out/sub1_killer_03.txt

@@ -0,0 +1 @@
+521296277

examples/testsets/abc129-f/out/sub1_killer_04.txt

@@ -0,0 +1 @@
+716070897

examples/testsets/abc129-f/out/sub1_killer_05.txt

@@ -0,0 +1 @@
+9086340

examples/testsets/abc129-f/out/sub1_killer_06.txt

@@ -0,0 +1 @@
+822961451

examples/testsets/abc129-f/out/sub1_killer_07.txt

@@ -0,0 +1 @@
+12975

examples/testsets/abc129-f/out/sub1_large_01.txt

@@ -0,0 +1 @@
+999999999

examples/testsets/abc129-f/out/sub1_large_02.txt

@@ -0,0 +1 @@
+739823859

examples/testsets/abc129-f/out/sub1_large_03.txt

@@ -0,0 +1 @@
+7215183

examples/testsets/abc129-f/out/sub1_large_04.txt

@@ -0,0 +1 @@
+795127085

examples/testsets/abc129-f/out/sub1_large_05.txt

@@ -0,0 +1 @@
+222138048

examples/testsets/abc129-f/out/sub1_large_06.txt

@@ -0,0 +1 @@
+96844045

examples/testsets/abc129-f/out/sub1_large_07.txt

@@ -0,0 +1 @@
+257000862

examples/testsets/abc129-f/out/sub1_large_08.txt

@@ -0,0 +1 @@
+55813670

examples/testsets/abc129-f/out/sub1_large_09.txt

@@ -0,0 +1 @@
+109152748

examples/testsets/abc129-f/out/sub1_rand_01.txt

@@ -0,0 +1 @@
+932512687

examples/testsets/abc129-f/out/sub1_rand_02.txt

@@ -0,0 +1 @@
+976788549

examples/testsets/abc129-f/out/sub1_rand_03.txt

@@ -0,0 +1 @@
+84300371

examples/testsets/abc129-f/out/sub1_rand_04.txt

@@ -0,0 +1 @@
+336830166

examples/testsets/abc129-f/out/sub1_rand_05.txt

@@ -0,0 +1 @@
+699837

examples/testsets/abc129-f/out/sub1_rand_06.txt

@@ -0,0 +1 @@
+184277614

examples/testsets/abc129-f/out/sub1_rand_07.txt

@@ -0,0 +1 @@
+357790932

examples/testsets/abc129-f/out/sub1_rand_08.txt

@@ -0,0 +1 @@
+878717921

examples/testsets/abc129-f/out/sub1_rand_09.txt

@@ -0,0 +1 @@
+207193062

examples/testsets/abc129-f/out/sub1_rand_10.txt

@@ -0,0 +1 @@
+336284961

examples/testsets/abc129-f/out/sub1_small_01.txt

@@ -0,0 +1 @@
+564963471

examples/testsets/abc129-f/out/sub1_small_02.txt

@@ -0,0 +1 @@
+997629525

examples/testsets/abc129-f/out/sub1_small_03.txt

@@ -0,0 +1 @@
+390889114

examples/testsets/abc129-f/out/sub1_small_04.txt

@@ -0,0 +1 @@
+792884

examples/testsets/abc129-f/out/sub1_small_05.txt

@@ -0,0 +1 @@
+203043563

examples/testsets/abc129-f/out/sub1_small_06.txt

@@ -0,0 +1 @@
+887967106