結果

問題 No.1085 桁和の桁和
ユーザー phspls
提出日時 2023-01-12 00:19:40
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,025 bytes
コンパイル時間 13,055 ms
コンパイル使用メモリ 394,448 KB
実行使用メモリ 12,416 KB
最終ジャッジ日時 2024-12-22 17:52:30
合計ジャッジ時間 15,759 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32 WA * 3
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable does not need to be mutable
  --> src/main.rs:21:9
   |
21 |     let mut cnt = t.iter().filter(|&&c| c == '?').count();
   |         ----^^^
   |         |
   |         help: remove this `mut`
   |
   = note: `#[warn(unused_mut)]` on by default

ソースコード

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();

    if d == 0 {
        if t.iter().any(|&c| c != '0' && c != '?') {
            println!("0");
        } else {
            println!("1");
        }
        return;
    }
    let summary = t.iter().filter(|&&c| c != '?').map(|&c| c as usize - '0' as usize).sum::<usize>() % 9;
    let mut cnt = t.iter().filter(|&&c| c == '?').count();
    let mut dp = vec![vec![0usize; 9]; cnt+1];
    dp[0][summary] = 1;
    for i in 0..cnt {
        for j in 0..9 {
            if dp[i][j] == 0 { continue; }
            for k in 0..10 {
                let jidx = (j+k)%9;
                dp[i+1][jidx] += dp[i][j];
                dp[i+1][jidx] %= MOD;
            }
        }
    }
    let d = d % 9;
    println!("{}", dp[cnt][d]);
}
0