結果

問題 No.2017 Mod7 Parade
ユーザー phspls
提出日時 2022-09-05 00:47:05
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 97 ms / 2,000 ms
コード長 1,901 bytes
コンパイル時間 14,941 ms
コンパイル使用メモリ 379,608 KB
実行使用メモリ 12,160 KB
最終ジャッジ日時 2024-11-19 08:31:44
合計ジャッジ時間 16,871 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

fn power(base: usize, times: usize, modval: usize) -> usize {
    if times == 0 { return 1usize; }
    if times == 1 { return base % modval; }
    let temp = power(base, times/2, modval);
    temp * temp % modval * power(base, times%2, modval) % modval
}

fn main() {
    let mut k = String::new();
    std::io::stdin().read_line(&mut k).ok();
    let k: usize = k.trim().parse().unwrap();
    let pairs = (0..k).map(|_| {
            let mut temp = String::new();
            std::io::stdin().read_line(&mut temp).ok();
            let temp: Vec<usize> = temp.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
            (temp[0], temp[1])
        })
        .collect::<Vec<(usize, usize)>>();

    let mut doubles = vec![vec![0usize; 10]; 32];
    let mut p2doubles = vec![0usize; 32];
    p2doubles[0] = 1;
    for i in 0..10 { doubles[0][i] = i%7; }
    for i in 0..31 {
        let slide = power(10usize, 1<<i, 7);
        p2doubles[i+1] = slide;
        for j in 0..10 {
            doubles[i+1][j] = doubles[i][j] * slide + doubles[i][j];
            doubles[i+1][j] %= 7;
        }
    }
    let mut dp = vec![vec![0usize; 7]; k+1];
    dp[0][0] = 1;
    for i in 0..k {
        let (d, l) = pairs[i];
        let mut targetval = 0usize;
        for j in 0..=30 {
            if ((l >> j) & 1) == 1 {
                targetval *= p2doubles[j+1];
                targetval %= 7;
                targetval += doubles[j][d];
                targetval %= 7;
            }
        }
        let shiftval = power(10usize, l, 7);
        for j in 0..7 {
            dp[i+1][j] += dp[i][j];
            dp[i+1][(j*shiftval+targetval)%7] += dp[i][j];
            dp[i+1][j] %= MOD;
            dp[i+1][(j*shiftval+targetval)%7] %= MOD;
        }
    }
    println!("{}", dp[k].iter().enumerate().skip(1).map(|(i, &v)| v*i).sum::<usize>() % MOD);
}
0