結果
問題 | No.1488 Max Score of the Tree |
ユーザー | stng |
提出日時 | 2021-09-07 20:23:46 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 300 ms / 2,000 ms |
コード長 | 1,162 bytes |
コンパイル時間 | 408 ms |
コンパイル使用メモリ | 87,168 KB |
実行使用メモリ | 154,328 KB |
最終ジャッジ日時 | 2023-08-25 12:46:57 |
合計ジャッジ時間 | 7,571 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 272 ms
150,728 KB |
testcase_01 | AC | 277 ms
148,000 KB |
testcase_02 | AC | 290 ms
151,836 KB |
testcase_03 | AC | 290 ms
154,084 KB |
testcase_04 | AC | 297 ms
154,260 KB |
testcase_05 | AC | 67 ms
71,272 KB |
testcase_06 | AC | 126 ms
95,300 KB |
testcase_07 | AC | 197 ms
120,688 KB |
testcase_08 | AC | 156 ms
107,228 KB |
testcase_09 | AC | 134 ms
97,992 KB |
testcase_10 | AC | 195 ms
122,216 KB |
testcase_11 | AC | 286 ms
154,328 KB |
testcase_12 | AC | 74 ms
76,656 KB |
testcase_13 | AC | 112 ms
88,132 KB |
testcase_14 | AC | 191 ms
113,284 KB |
testcase_15 | AC | 141 ms
99,084 KB |
testcase_16 | AC | 87 ms
80,220 KB |
testcase_17 | AC | 114 ms
90,296 KB |
testcase_18 | AC | 221 ms
128,084 KB |
testcase_19 | AC | 152 ms
104,844 KB |
testcase_20 | AC | 108 ms
87,532 KB |
testcase_21 | AC | 88 ms
81,632 KB |
testcase_22 | AC | 137 ms
98,460 KB |
testcase_23 | AC | 67 ms
71,412 KB |
testcase_24 | AC | 68 ms
71,156 KB |
testcase_25 | AC | 68 ms
71,260 KB |
testcase_26 | AC | 117 ms
93,232 KB |
testcase_27 | AC | 88 ms
75,956 KB |
testcase_28 | AC | 99 ms
84,232 KB |
testcase_29 | AC | 106 ms
87,172 KB |
testcase_30 | AC | 245 ms
139,012 KB |
testcase_31 | AC | 300 ms
154,192 KB |
ソースコード
import sys sys.setrecursionlimit(10**6) n,k = map(int,input().split()) ab = [[] for i in range(n)] val = [] for i in range(n-1): a,b,c = map(int,input().split()) a -= 1 b -= 1 ab[a].append([b,i]) ab[b].append([a,i]) val.append(c) li = [0]*(n-1) def dfs(v,pre,idx): tmp = 0 for i in range(len(ab[v])): if ab[v][i][0] != pre: #print(ab[v][i]+1) dfs(ab[v][i][0],v,ab[v][i][1]) tmp += li[ab[v][i][1]] #print(ab[v],len(ab[v]),"len") if i == len(ab[v])-1: tmp -= 1 tmp = max(0,tmp) #print(v+1) li[idx] = tmp+1 #print(li,idx) #return tmp+1 for v,idx in ab[0]: #print(v,idx) dfs(v,0,idx) #print(li,"li") dp = [[0]*(k+1) for _ in range(n)] ans = 0 for i in range(n-1): ans += val[i]*li[i] for i in range(n-1): for j in range(k+1): if j-val[i] >= 0: dp[i+1][j] = max(dp[i][j-val[i]]+val[i]*li[i],dp[i][j]) else: dp[i+1][j] = dp[i][j] #for i in range(k+1): # add = max(add,dp[n-1][i]) #print(dp) add = dp[n-1][k] print(add+ans)