結果

問題 No.372 It's automatic
ユーザー akakimidori
提出日時 2021-03-17 15:05:06
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 968 ms / 6,000 ms
コード長 913 bytes
コンパイル時間 13,442 ms
コンパイル使用メモリ 382,948 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-15 00:56:48
合計ジャッジ時間 25,324 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

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 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];
        for (i, dp) in dp.iter().enumerate() {
            next[(10 * i + k) % m] += *dp;
        }
        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