結果

問題 No.1111 コード進行
ユーザー fukafukatani
提出日時 2020-10-10 11:07:34
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 159 ms / 2,000 ms
コード長 1,163 bytes
コンパイル時間 14,599 ms
コンパイル使用メモリ 378,860 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 15:43:41
合計ジャッジ時間 15,183 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 48
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused variable: `i`
 --> src/main.rs:5:9
  |
5 |     for i in 0..m {
  |         ^ help: if this is intentional, prefix it with an underscore: `_i`
  |
  = note: `#[warn(unused_variables)]` on by default

ソースコード

diff #

fn main() {
    let v = read_vec::<usize>();
    let (n, m, k) = (v[0], v[1], v[2]);
    let mut codes = vec![];
    for i in 0..m {
        let v = read_vec::<usize>();
        let (p, q, c) = (v[0] - 1, v[1] - 1, v[2]);
        codes.push((p, q, c));
    }
    let mut dp = vec![vec![0i64; k + 1]; 300];

    for i in 0..300 {
        dp[i][0] = 1;
    }

    for _ in 0..n - 1 {
        let mut next = vec![vec![0i64; k + 1]; 300];

        for &(p, q, c) in &codes {
            for prev_k in 0..k + 1 {
                if prev_k + c > k {
                    continue;
                }
                next[q][prev_k + c] += dp[p][prev_k];
                next[q][prev_k + c] %= 1000000007;
            }
        }
        dp = next;
    }
    let ans = dp.iter().map(|x| x[k]).fold(0, |s, x| (s + x) % 1000000007);
    println!("{}", ans);
}

fn read<T: std::str::FromStr>() -> T {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().parse().ok().unwrap()
}

fn read_vec<T: std::str::FromStr>() -> Vec<T> {
    read::<String>()
        .split_whitespace()
        .map(|e| e.parse().ok().unwrap())
        .collect()
}
0