結果
問題 | No.1488 Max Score of the Tree |
ユーザー |
![]() |
提出日時 | 2021-04-23 22:27:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 296 ms / 2,000 ms |
コード長 | 996 bytes |
コンパイル時間 | 183 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 142,848 KB |
最終ジャッジ日時 | 2024-07-04 08:18:15 |
合計ジャッジ時間 | 6,042 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
def tree_dfs(g, root=0): s = [root] d = [True] * len(g) d[root] = False order = [] while s: p = s.pop() order.append(p) d[p] = False for node, _ in g[p]: if d[node]: s.append(node) return order n, k = map(int, input().split()) g = [[] for _ in range(n)] for i in range(n - 1): a, b, c = map(int, input().split()) g[a - 1].append((b - 1, c)) g[b - 1].append((a - 1, c)) order = tree_dfs(g) d = [-1] * n a = [0] * n b = [0] * n for v in order[::-1]: d[v] = 0 for node, cost in g[v]: if d[node] != -1: d[v] += d[node] a[node] += d[node] * cost b[node] = cost if d[v] == 0: d[v] = 1 dp = [[0] * (k + 1) for _ in range(n + 1)] for i in range(n): for j in range(k + 1): if j - b[i] >= 0: dp[i + 1][j] = max(dp[i][j - b[i]] + a[i], dp[i][j]) else: dp[i + 1][j] = dp[i][j] print(max(dp[n]) + sum(a))