結果

問題 No.1407 Kindness
ユーザー phspls
提出日時 2023-01-22 21:47:22
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 300 ms / 2,000 ms
コード長 1,402 bytes
コンパイル時間 12,942 ms
コンパイル使用メモリ 401,704 KB
実行使用メモリ 57,472 KB
最終ジャッジ日時 2024-06-25 00:10:47
合計ジャッジ時間 18,223 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable `SIZE` should have a snake case name
 --> src/main.rs:9:9
  |
9 |     let SIZE = n.len();
  |         ^^^^ help: convert the identifier to snake case: `size`
  |
  = note: `#[warn(non_snake_case)]` on by default

ソースコード

diff # 

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

fn main() {
    let mut n = String::new();
    std::io::stdin().read_line(&mut n).ok();
    let n = n.trim().chars().map(|c| c as usize - '0' as usize).collect::<Vec<_>>();

    let SIZE = n.len();
    let mut edp = vec![vec![vec![0usize; 10]; 2]; SIZE];
    let mut ldp = vec![vec![vec![0usize; 10]; 2]; SIZE];
    edp[0][0][n[0]] = n[0];
    for i in 0..n[0] {
        ldp[0][if i == 0 { 1 } else { 0 }][i] = i.max(1);
    }
    for i in 1..SIZE {
        for leading0 in 0..2 {
            for j in 0..10 {
                for nj in 0..10 {
                    let nleading0 = if leading0 == 1 && nj == 0 { 1 } else { 0 };
                    let val = if nleading0 == 1 { 1 } else { nj };
                    if n[i] == nj {
                        edp[i][nleading0][nj] += edp[i-1][leading0][j] * val % MOD;
                        edp[i][nleading0][nj] %= MOD;
                    } else if nj < n[i] {
                        ldp[i][nleading0][nj] += edp[i-1][leading0][j] * val % MOD;
                    }
                    ldp[i][nleading0][nj] += ldp[i-1][leading0][j] * val % MOD;
                    ldp[i][nleading0][nj] %= MOD;
                }
            }
        }
    }
    println!("{}", (edp[SIZE-1].iter().map(|v| v.iter().sum::<usize>()).sum::<usize>() + ldp[SIZE-1].iter().map(|v| v.iter().sum::<usize>()).sum::<usize>() - 1) % MOD);
}
0