結果

問題 No.1111 コード進行
ユーザー Strorkis
提出日時 2020-07-10 22:42:00
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 283 ms / 2,000 ms
コード長 1,441 bytes
コンパイル時間 13,291 ms
コンパイル使用メモリ 402,024 KB
実行使用メモリ 216,320 KB
最終ジャッジ日時 2025-02-07 15:30:37
合計ジャッジ時間 15,138 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #

fn main() {
    const MOD: usize = 1_000_000_007;
    const CN: usize = 300;

    let (n, m, k): (usize, usize, usize) = {
        let mut buf = String::new();
        std::io::stdin().read_line(&mut buf).unwrap();
        let mut iter = buf.split_whitespace();
        (
            iter.next().unwrap().parse().unwrap(),
            iter.next().unwrap().parse().unwrap(),
            iter.next().unwrap().parse().unwrap(),
        )
    };

    let mut p = Vec::with_capacity(m);
    let mut q = Vec::with_capacity(m);
    let mut c = Vec::with_capacity(m);
    for _ in 0..m {
        let mut buf = String::new();
        std::io::stdin().read_line(&mut buf).unwrap();
        let mut iter = buf.split_whitespace();
        p.push(iter.next().unwrap().parse::<usize>().unwrap() - 1);
        q.push(iter.next().unwrap().parse::<usize>().unwrap() - 1);
        c.push(iter.next().unwrap().parse::<usize>().unwrap());
    }

    let mut dp = vec![vec![vec![0; k + 1]; CN]; n];
    for i in 0..CN {
        dp[0][i][0] = 1;
    }
    for i in 0..(n - 1) {
        for j in 0..m {
            for l in 0..=k {
                if l + c[j] > k { continue; }
                dp[i + 1][q[j]][l + c[j]] = {
                    (dp[i + 1][q[j]][l + c[j]] + dp[i][p[j]][l]) % MOD
                };
            }
        }
    }

    let mut ans = 0;
    for i in 0..CN {
        ans = (ans + dp[n - 1][i][k]) % MOD;
    }
    println!("{}", ans);
}
0