結果
問題 | No.1488 Max Score of the Tree |
ユーザー |
|
提出日時 | 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
ソースコード
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]