結果

問題 No.1417 100の倍数かつ正整数(2)
コンテスト
ユーザー phspls
提出日時 2022-11-16 00:25:46
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 105 ms / 3,000 ms
コード長 1,867 bytes
コンパイル時間 14,330 ms
コンパイル使用メモリ 386,004 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2024-09-16 22:45:20
合計ジャッジ時間 12,947 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

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 limit = n.len();
    let cnt0 = vec![1, 0, 0, 0, 0, 0, 0, 0, 0, 0];
    let cnt2 = vec![0, 0, 1, 0, 2, 0, 1, 0, 2, 0];
    let cnt5 = vec![0, 0, 0, 0, 0, 1, 0, 0, 0, 0];
    let mut edp = vec![vec![vec![vec![vec![0usize; 3]; 3]; 2]; 2]; limit];
    let mut ldp = vec![vec![vec![vec![vec![0usize; 3]; 3]; 2]; 2]; limit];
    edp[0][0][0][cnt2[n[0]]][cnt5[n[0]]] = 1;
    for i in 0..n[0] {
        ldp[0][cnt0[i]][0][cnt2[i]][cnt5[i]] += 1;
    }
    for i in 1..limit {
        for lead0 in 0..2 {
            for cons0 in 0..2 {
                for div2 in 0..3 {
                    for div5 in 0..3 {
                        for j in 0..10 {
                            let nlead0 = if lead0 == 1 && j == 0 { 1 } else { 0 };
                            let ncons0 = if cons0 == 1 || (lead0 == 0 && j == 0) { 1 } else { 0 };
                            let ndiv2 = 2.min(div2 + cnt2[j]);
                            let ndiv5 = 2.min(div5 + cnt5[j]);
                            if j == n[i] {
                                edp[i][nlead0][ncons0][ndiv2][ndiv5] += edp[i-1][lead0][cons0][div2][div5];
                            } else if j < n[i] {
                                ldp[i][nlead0][ncons0][ndiv2][ndiv5] += edp[i-1][lead0][cons0][div2][div5];
                            }
                            ldp[i][nlead0][ncons0][ndiv2][ndiv5] += ldp[i-1][lead0][cons0][div2][div5];
                            ldp[i][nlead0][ncons0][ndiv2][ndiv5] %= MOD;
                        }
                    }
                }
            }
        }
    }
    println!("{}", (edp[limit-1][0][0][2][2] + ldp[limit-1][0][0][2][2]) % MOD);
}
0