結果

問題 No.372 It's automatic
コンテスト
ユーザー akakimidori
提出日時 2021-03-17 15:06:46
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 642 ms / 6,000 ms
コード長 1,041 bytes
コンパイル時間 17,417 ms
コンパイル使用メモリ 390,988 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-15 00:56:21
合計ジャッジ時間 24,477 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused variable: `i`
  --> src/main.rs:20:14
   |
20 |         for (i, dp) in dp.iter().enumerate() {
   |              ^ help: if this is intentional, prefix it with an underscore: `_i`
   |
   = note: `#[warn(unused_variables)]` on by default

ソースコード

diff # 

fn read() -> (Vec<u8>, usize) {
    let mut s = String::new();
    use std::io::*;
    std::io::stdin().read_to_string(&mut s).unwrap();
    let mut it = s.trim().split_whitespace();
    let s: Vec<u8> = it.next().unwrap().bytes().collect();
    (s, it.next().unwrap().parse().unwrap())
}

fn main() {
    let (s, m) = read();
    const MOD: u64 = 1_000_000_007;
    let add = 10 % m;
    let mut ans = 0;
    let mut dp = vec![0; m];
    for k in s {
        let k = (k - b'0') as usize;
        let mut next = vec![0; m];
        let mut pos = k % m;
        for (i, dp) in dp.iter().enumerate() {
            next[pos] += *dp;
            pos += add;
            if pos >= m {
                pos -= m;
            }
        }
        if k > 0 {
            next[k % m] += 1;
        }
        ans += next[0] + (k == 0) as u64;
        for (next, dp) in next.iter_mut().zip(dp.iter()) {
            *next += *dp;
        }
        dp = next;
        dp.iter_mut().for_each(|p| *p %= MOD);
    }
    ans %= MOD;
    println!("{}", ans);
}
0