結果

問題 No.801 エレベーター
ユーザー phspls
提出日時 2022-12-16 23:41:32
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 115 ms / 2,000 ms
コード長 1,356 bytes
コンパイル時間 12,534 ms
コンパイル使用メモリ 389,728 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-16 05:39:56
合計ジャッジ時間 15,784 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

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

fn main() {
    let mut nmk = String::new();
    std::io::stdin().read_line(&mut nmk).ok();
    let nmk: Vec<usize> = nmk.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
    let n = nmk[0];
    let m = nmk[1];
    let k = nmk[2];
    let items = (0..m).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]-1, temp[1]-1)
        })
        .collect::<Vec<_>>();

    let mut dp = vec![0usize; n];
    dp[0] = 1;
    for _ in 0..k {
        let mut prev_sum = vec![0usize; n+1];
        for i in 0..n {
            prev_sum[i+1] = prev_sum[i] + dp[i];
            prev_sum[i+1] %= MOD;
        }
        let mut next_sum = vec![0usize; n+1];
        for &(l, r) in items.iter() {
            let total = MOD + prev_sum[r+1] - prev_sum[l];
            let total = total % MOD;
            next_sum[l] += total;
            next_sum[l] %= MOD;
            next_sum[r+1] += MOD - total;
            next_sum[r+1] %= MOD;
        }
        for i in 0..n {
            next_sum[i+1] += next_sum[i];
            next_sum[i+1] %= MOD;
        }
        next_sum.pop();
        dp = next_sum;
    }
    println!("{}", dp[n-1]);
}
0