結果

問題 No.2017 Mod7 Parade
ユーザー phsplsphspls
提出日時 2022-09-05 00:47:05
言語 Rust
(1.77.0)
結果
AC  
実行時間 95 ms / 2,000 ms
コード長 1,901 bytes
コンパイル時間 14,425 ms
コンパイル使用メモリ 378,432 KB
実行使用メモリ 12,160 KB
最終ジャッジ日時 2024-04-29 23:31:27
合計ジャッジ時間 16,404 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 0 ms
5,376 KB
testcase_03 AC 74 ms
9,856 KB
testcase_04 AC 49 ms
7,168 KB
testcase_05 AC 38 ms
5,760 KB
testcase_06 AC 82 ms
10,752 KB
testcase_07 AC 11 ms
5,376 KB
testcase_08 AC 46 ms
6,784 KB
testcase_09 AC 18 ms
5,376 KB
testcase_10 AC 45 ms
6,912 KB
testcase_11 AC 65 ms
8,832 KB
testcase_12 AC 57 ms
7,936 KB
testcase_13 AC 94 ms
12,160 KB
testcase_14 AC 95 ms
12,032 KB
testcase_15 AC 95 ms
12,032 KB
testcase_16 AC 95 ms
12,032 KB
testcase_17 AC 95 ms
12,032 KB
testcase_18 AC 84 ms
12,032 KB
testcase_19 AC 34 ms
12,032 KB
testcase_20 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

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