結果

問題 No.1488 Max Score of the Tree
ユーザー magurofly
提出日時 2021-04-23 22:03:21
言語 Ruby
(3.4.1)
結果
AC  
実行時間 1,539 ms / 2,000 ms
コード長 829 bytes
コンパイル時間 484 ms
コンパイル使用メモリ 7,296 KB
実行使用メモリ 70,784 KB
最終ジャッジ日時 2024-07-04 08:04:16
合計ジャッジ時間 24,327 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

N, K = gets.split.map(&:to_i)
graph = Array.new(N) { [] }
weight = []
(N - 1).times do
    a, b, c = gets.split.map(&:to_i)
    e = weight.size
    graph[a - 1] << [b - 1, e]
    graph[b - 1] << [a - 1, e]
    weight << c
end

value = [0] * (N - 1)
path = []
visited = [false] * N
dfs = ->(u) {
    visited[u] = true
    has_children = false
    graph[u].each do |(v, e)|
        next if visited[v]
        has_children = true
        path << e
        dfs[v]
        path.pop
    end
    unless has_children
      path.each do |e|
        value[e] += weight[e]
      end
    end
}
dfs[0]

dp = Array.new(K + 1, 0)
(0 ... N - 1).each do |i|
  dp2 = Array.new(K + 1, 0)
  (0 .. K).each do |w|
    v = dp[w]
    v = [v, dp[w - weight[i]] + value[i]].max if w >= weight[i]
    dp2[w] = v
  end
  dp = dp2
end

puts value.sum + dp[K]
0