結果

問題 No.1417 100の倍数かつ正整数(2)
コンテスト
ユーザー phspls
提出日時 2022-12-06 21:17:19
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 100 ms / 3,000 ms
コード長 1,856 bytes
コンパイル時間 16,039 ms
コンパイル使用メモリ 404,044 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2024-10-13 09:13:43
合計ジャッジ時間 17,504 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 mut edp = vec![vec![vec![vec![vec![0usize; 3]; 3]; 2]; 2]; n.len()];
    let mut ldp = vec![vec![vec![vec![vec![0usize; 3]; 3]; 2]; 2]; n.len()];
    let mul2 = vec![0, 0, 1, 0, 2, 0, 1, 0, 2, 0];
    let mul5 = vec![0, 0, 0, 0, 0, 1, 0, 0, 0, 0];
    edp[0][0][0][mul5[n[0]]][mul2[n[0]]] = 1;
    for i in 0..n[0] {
        ldp[0][if i == 0 { 1 } else { 0 }][0][mul5[i]][mul2[i]] += 1;
    }
    for i in 1..n.len() {
        for lead0 in 0..2 {
            for inc0 in 0..2 {
                for multi5 in 0..3 {
                    for multi2 in 0..3 {
                        for j in 0..10 {
                            let nlead0 = if lead0 == 1 && j == 0 { 1 } else { 0 };
                            let ninc0 = if inc0 == 1 || (lead0 == 0 && j == 0) { 1 } else { 0 };
                            let nmulti5 = 2.min(multi5 + mul5[j]);
                            let nmulti2 = 2.min(multi2 + mul2[j]);
                            if n[i] == j {
                                edp[i][nlead0][ninc0][nmulti5][nmulti2] += edp[i-1][lead0][inc0][multi5][multi2];
                            } else if j < n[i] {
                                ldp[i][nlead0][ninc0][nmulti5][nmulti2] += edp[i-1][lead0][inc0][multi5][multi2];
                            }
                            ldp[i][nlead0][ninc0][nmulti5][nmulti2] += ldp[i-1][lead0][inc0][multi5][multi2];
                            ldp[i][nlead0][ninc0][nmulti5][nmulti2] %= MOD;
                        }
                    }
                }
            }
        }
    }
    println!("{}", (MOD + edp[n.len()-1][0][0][2][2] + ldp[n.len()-1][0][0][2][2]) % MOD);
}
0