結果

問題 No.763 Noelちゃんと木遊び
ユーザー Kohei AsanoKohei Asano
提出日時 2021-02-04 20:14:36
言語 Rust
(1.77.0)
結果
RE  
実行時間 -
コード長 2,370 bytes
コンパイル時間 2,637 ms
コンパイル使用メモリ 156,164 KB
実行使用メモリ 8,956 KB
最終ジャッジ日時 2023-09-13 16:41:23
合計ジャッジ時間 4,660 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: function `solve` is never used
 --> Main.rs:1:4
  |
1 | fn solve() {
  |    ^^^^^
  |
  = note: `#[warn(dead_code)]` on by default

warning: 1 warning emitted

ソースコード

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