結果

問題 No.417 チューリップバブル
ユーザー Kohei Asano
提出日時 2021-02-04 20:16:16
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 743 ms / 2,000 ms
コード長 2,370 bytes
コンパイル時間 14,950 ms
コンパイル使用メモリ 400,764 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-01 01:26:56
合計ジャッジ時間 22,595 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: function `solve` is never used
 --> src/main.rs:1:4
  |
1 | fn solve() {
  |    ^^^^^
  |
  = note: `#[warn(dead_code)]` on by default

ソースコード

diff # 

fn solve() {
    let (a, b, c): (usize, usize, usize) = (read(), read(), read());
    println!("{:?}", a + b + c);
    let s: String = read();
    println!("{:?}", s);
}

fn dfs(
    u: usize,
    p: usize,
    time_to_come: usize,
    adjl: &Vec<Vec<(usize, usize)>>,
    dp: &mut Vec<Vec<usize>>,
    m: usize,
    zei: &Vec<usize>,
) {
    for &(v, c) in &adjl[u] {
        if v == p {
            continue;
        }
        if (time_to_come + c) * 2 <= m {
            dfs(v, u, time_to_come + c, adjl, dp, m, zei);
            // 部分木の最適解が求まったとする.
            // その子に行く場合
            // b時間uで使う時間の最適値は求まってなさそう.
            for b in (0..m - c * 2 + 1).rev() {
                for t in 0..m - c * 2 - b + 1 {
                    dp[u][b + 2 * c + t] = max(dp[u][b + 2 * c + t], dp[v][t] + dp[u][b]);
                }
            }
        }
    }
}

fn main() {
    let (n, m): (usize, usize) = (read::<usize>(), read::<usize>());
    let zei: Vec<usize> = (0..n).map(|_| read::<usize>()).collect();
    let mut adjl: Vec<Vec<(usize, usize)>> = vec![vec![]; n];
    for _ in 0..n - 1 {
        let (a, b, c): (usize, usize, usize) = (read::<usize>(), read::<usize>(), read::<usize>());
        adjl[a].push((b, c));
        adjl[b].push((a, c));
    }
    // dp[i][t] ... iでt時間使って集められる税の最大値.
    let mut dp = vec![vec![0; m + 1]; n];
    for i in 0..n {
        for j in 0..m + 1 {
            dp[i][j] = zei[i];
        }
    }
    dfs(0, 0, 0, &adjl, &mut dp, m, &zei);
    let mut res = 0;
    for i in 0..m + 1 {
        res = max(res, dp[0][i]);
    }
    println!("{}", res);
}

// =========
#[allow(unused_imports)]
use std::cmp::{max, min, Reverse};
#[allow(unused_imports)]
use std::collections::{BinaryHeap, HashMap, HashSet};
#[allow(unused_imports)]
use std::process::exit;

#[allow(dead_code)]
const MOD: usize = 1000000007;

fn read<T: std::str::FromStr>() -> T {
    use std::io::Read;
    let stdin = std::io::stdin();
    let stdin = stdin.lock();
    let token: String = stdin
        .bytes()
        .map(|c| c.expect("failed to read char") as char)
        .skip_while(|c| c.is_whitespace())
        .take_while(|c| !c.is_whitespace())
        .collect();
    token.parse().ok().expect("failed to parse token")
}

// =========
0