結果

問題 No.372 It's automatic
ユーザー akakimidoriakakimidori
提出日時 2021-03-17 15:08:51
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 642 ms / 6,000 ms
コード長 1,094 bytes
コンパイル時間 12,727 ms
コンパイル使用メモリ 387,892 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-15 00:58:26
合計ジャッジ時間 21,664 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 add = 10 % m;
    let mut ans = 0;
    let mut dp = vec![0; m];
    let mut next = vec![0; m];
    for k in s {
        let k = (k - b'0') as usize;
        next.clear();
        next.resize(m, 0);
        let mut pos = k % m;
        for dp in dp.iter() {
            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;
        }
        std::mem::swap(&mut dp, &mut next);
        dp.iter_mut().for_each(|p| *p %= MOD);
    }
    ans %= MOD;
    println!("{}", ans);
}
0