結果

問題 No.1417 100の倍数かつ正整数(2)
ユーザー phspls
提出日時 2022-11-09 02:01:29
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 105 ms / 3,000 ms
コード長 1,869 bytes
コンパイル時間 12,632 ms
コンパイル使用メモリ 391,460 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2024-07-22 09:20:16
合計ジャッジ時間 14,665 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
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 map0 = vec![1,0,0,0,0,0,0,0,0,0];
    let map2 = vec![0,0,1,0,2,0,1,0,2,0];
    let map5 = vec![0,0,0,0,0,1,0,0,0,0];
    // 桁 -> leading 0か -> 0があるか -> 2の個数 -> 5の個数
    let mut edp = vec![vec![vec![vec![vec![0usize; 3]; 3]; 2]; 2]; SIZE];
    let mut ldp = vec![vec![vec![vec![vec![0usize; 3]; 3]; 2]; 2]; SIZE];
    edp[0][0][0][map2[n[0]]][map5[n[0]]] = 1;
    for i in 0..n[0] {
        ldp[0][map0[i]][0][map2[i]][map5[i]] += 1;
    }
    for i in 1..SIZE {
        for leading0 in 0..2 {
            for j0 in 0..2 {
                for j2 in 0..3 {
                    for j5 in 0..3 {
                        for j in 0..10 {
                            let nleading0 = if leading0 == 1 && j == 0 { 1 } else { 0 };
                            let nj0 = if nleading0 == 1 || (j0 == 0 && j > 0) { 0 } else { 1 };
                            let nj2 = (j2+map2[j]).min(2);
                            let nj5 = (j5+map5[j]).min(2);
                            if j == n[i] {
                                edp[i][nleading0][nj0][nj2][nj5] += edp[i-1][leading0][j0][j2][j5];
                            } else if j < n[i] {
                                ldp[i][nleading0][nj0][nj2][nj5] += edp[i-1][leading0][j0][j2][j5];
                            }
                            ldp[i][nleading0][nj0][nj2][nj5] += ldp[i-1][leading0][j0][j2][j5];
                            ldp[i][nleading0][nj0][nj2][nj5] %= MOD;
                        }
                    }
                }
            }
        }
    }
    println!("{}", (edp[SIZE-1][0][0][2][2] + ldp[SIZE-1][0][0][2][2]) % MOD);
}
0