結果

問題 No.1417 100の倍数かつ正整数(2)
ユーザー phsplsphspls
提出日時 2022-11-09 23:52:40
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 115 ms / 3,000 ms
コード長 1,933 bytes
コンパイル時間 13,545 ms
コンパイル使用メモリ 395,452 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2024-07-23 04:41:45
合計ジャッジ時間 15,125 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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];
    // n -> leading0 -> 0 included -> 2 times -> 5 times
    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 inc0 in 0..2 {
                for time2 in 0..3 {
                    for time5 in 0..3 {
                        for j in 0..10 {
                            let nleading0 = if leading0 == 1 && j == 0 { 1 } else { 0 };
                            let ninc0 = if leading0 == 1 || (inc0 == 0 && j > 0) { 0 } else { 1 };
                            let ntime2 = (time2+map2[j]).min(2);
                            let ntime5 = (time5+map5[j]).min(2);
                            if j == n[i] {
                                edp[i][nleading0][ninc0][ntime2][ntime5] += edp[i-1][leading0][inc0][time2][time5];
                            } else if j < n[i] {
                                ldp[i][nleading0][ninc0][ntime2][ntime5] += edp[i-1][leading0][inc0][time2][time5];
                            }
                            ldp[i][nleading0][ninc0][ntime2][ntime5] += ldp[i-1][leading0][inc0][time2][time5];
                            ldp[i][nleading0][ninc0][ntime2][ntime5] %= MOD;
                        }
                    }
                }
            }
        }
    }
    println!("{}", (edp[SIZE-1][0][0][2][2] + ldp[SIZE-1][0][0][2][2]) % MOD);
}
0