結果
問題 | No.1488 Max Score of the Tree |
ユーザー | rlangevin |
提出日時 | 2023-02-21 01:04:27 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 294 ms / 2,000 ms |
コード長 | 1,303 bytes |
コンパイル時間 | 300 ms |
コンパイル使用メモリ | 87,172 KB |
実行使用メモリ | 155,348 KB |
最終ジャッジ日時 | 2023-09-28 19:40:01 |
合計ジャッジ時間 | 8,172 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 270 ms
151,796 KB |
testcase_01 | AC | 272 ms
148,604 KB |
testcase_02 | AC | 288 ms
152,528 KB |
testcase_03 | AC | 289 ms
155,348 KB |
testcase_04 | AC | 294 ms
154,984 KB |
testcase_05 | AC | 101 ms
71,384 KB |
testcase_06 | AC | 152 ms
95,980 KB |
testcase_07 | AC | 213 ms
121,508 KB |
testcase_08 | AC | 177 ms
107,976 KB |
testcase_09 | AC | 162 ms
98,748 KB |
testcase_10 | AC | 208 ms
122,912 KB |
testcase_11 | AC | 283 ms
155,288 KB |
testcase_12 | AC | 109 ms
77,252 KB |
testcase_13 | AC | 136 ms
77,036 KB |
testcase_14 | AC | 197 ms
114,100 KB |
testcase_15 | AC | 165 ms
99,944 KB |
testcase_16 | AC | 117 ms
80,900 KB |
testcase_17 | AC | 142 ms
91,144 KB |
testcase_18 | AC | 226 ms
128,896 KB |
testcase_19 | AC | 174 ms
105,624 KB |
testcase_20 | AC | 137 ms
88,296 KB |
testcase_21 | AC | 123 ms
82,344 KB |
testcase_22 | AC | 163 ms
99,112 KB |
testcase_23 | AC | 98 ms
71,280 KB |
testcase_24 | AC | 98 ms
71,624 KB |
testcase_25 | AC | 101 ms
71,516 KB |
testcase_26 | AC | 147 ms
94,040 KB |
testcase_27 | AC | 117 ms
77,064 KB |
testcase_28 | AC | 130 ms
85,024 KB |
testcase_29 | AC | 136 ms
87,672 KB |
testcase_30 | AC | 248 ms
139,580 KB |
testcase_31 | AC | 293 ms
155,272 KB |
ソースコード
from collections import deque def non_rec_dfs(s): stack = [] stack.append(~s) stack.append(s) par = [-1] * N dist = [0] * N inf = 10 ** 18 pre = [-inf] * (K + 1) pre[0] = 0 leaf = [0] * N while stack: u = stack.pop() if u >= 0: stack.append(~u) flag = 0 for v, c in G[u]: if v == par[u]: continue par[v] = u dist[v] = dist[u] + c flag = 1 stack.append(v) if not flag: leaf[u] = 1 else: u = ~u if par[u] == -1: continue leaf[par[u]] += leaf[u] dp = [-inf] * (K + 1) cost = D[(par[u], u)] for i in range(K + 1): dp[i] = pre[i] + cost * leaf[u] if i - cost >= 0: dp[i] = max(dp[i], pre[i - cost] + 2 * cost * leaf[u]) dp, pre = pre, dp return max(pre) N, K = map(int, input().split()) G = [[] for i in range(N)] D = dict() for i in range(N - 1): a, b, c = map(int, input().split()) a, b = a - 1, b - 1 G[a].append((b, c)) G[b].append((a, c)) D[(a, b)] = c D[(b, a)] = c print(non_rec_dfs(0))