結果

問題 No.1488 Max Score of the Tree
ユーザー StrorkisStrorkis
提出日時 2021-04-23 22:49:03
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 17 ms / 2,000 ms
コード長 1,461 bytes
コンパイル時間 13,196 ms
コンパイル使用メモリ 384,592 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-04 08:31:49
合計ジャッジ時間 14,246 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

fn dfs(
    g: &[Vec<(usize, usize)>], cnt: &mut [usize],
    prev: usize, from: usize,
) -> usize {
    let mut res = 0;
    for &(to, i) in &g[from] {
        if to == prev { continue; }
        let x = dfs(g, cnt, from, to);
        cnt[i] += x;
        res += x;
    }
    res.max(1)
}

fn main() {
    let ref mut buf = String::new();
    std::io::Read::read_to_string(&mut std::io::stdin(), buf).ok();
    let mut iter = buf.split_whitespace();

    macro_rules! scan {
        ([$t:tt; $n:expr]) => ((0..$n).map(|_| scan!($t)).collect::<Vec<_>>());
        (($($t:tt),*)) => (($(scan!($t)),*));
        (Usize1) => (scan!(usize) - 1);
        (Bytes) => (scan!(String).into_bytes());
        ($t:ty) => (iter.next().unwrap().parse::<$t>().unwrap());
    }

    let (n, k) = scan!((usize, usize));

    let mut g = vec![vec![]; n];
    let mut l = vec![0; n - 1];
    for i in 0..(n - 1) {
        let (a, b, c) = scan!((Usize1, Usize1, usize));
        g[a].push((b, i));
        g[b].push((a, i));
        l[i] = c;
    }

    let mut cnt = vec![0; n - 1];
    dfs(&g, &mut cnt, !0, 0);

    let mut ans = 0;
    let mut dp = vec![0; k + 1];
    dp[0] = 0;
    for (l, cnt) in l.into_iter().zip(cnt.into_iter()) {
        ans += l * cnt;
        for i in (0..=k).rev() {
            if i + l <= k {
                dp[i + l] = dp[i + l].max(dp[i] + l * cnt);
            }
        }
    }

    ans += dp.iter().max().unwrap();
    println!("{}", ans);
}
0