結果

問題 No.1085 桁和の桁和
コンテスト
ユーザー phspls
提出日時 2023-01-19 04:47:26
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 48 ms / 2,000 ms
コード長 1,110 bytes
コンパイル時間 22,217 ms
コンパイル使用メモリ 384,192 KB
実行使用メモリ 12,672 KB
最終ジャッジ日時 2025-01-24 09:02:25
合計ジャッジ時間 19,600 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

const MOD: usize = 1e9 as usize + 7;

fn main() {
    let mut t = String::new();
    std::io::stdin().read_line(&mut t).ok();
    let t = t.trim().chars().collect::<Vec<_>>();
    let mut d = String::new();
    std::io::stdin().read_line(&mut d).ok();
    let d: usize = d.trim().parse().unwrap();

    let n = t.len();
    let avail0 = t.iter().all(|&c| c == '?' || c == '0');
    if d == 0 {
        if avail0 {
            println!("1");
        } else {
            println!("0");
        }
        return;
    }
    let mut dp = vec![vec![0usize; 9]; n+1];
    dp[0][0] = 1;
    for i in 0..n {
        for j in 0..9 {
            if t[i] != '?' {
                let idx = (j + t[i] as usize - '0' as usize) % 9;
                dp[i+1][idx] += dp[i][j];
                dp[i+1][idx] %= MOD;
            } else {
                for k in 0..10 {
                    let idx = (j + k) % 9;
                    dp[i+1][idx] += dp[i][j];
                    dp[i+1][idx] %= MOD;
                }
            }
        }
    }
    println!("{}", if d == 9 && avail0 { dp[n][0] - 1 } else { dp[n][d%9] });
}
0