結果
問題 | No.1488 Max Score of the Tree |
ユーザー | mkawa2 |
提出日時 | 2021-04-23 22:41:43 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,331 ms / 2,000 ms |
コード長 | 1,233 bytes |
コンパイル時間 | 314 ms |
コンパイル使用メモリ | 10,956 KB |
実行使用メモリ | 11,480 KB |
最終ジャッジ日時 | 2023-09-17 12:45:09 |
合計ジャッジ時間 | 14,198 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge11 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 547 ms
10,328 KB |
testcase_01 | AC | 1,331 ms
11,480 KB |
testcase_02 | AC | 1,203 ms
11,100 KB |
testcase_03 | AC | 991 ms
10,956 KB |
testcase_04 | AC | 1,327 ms
11,460 KB |
testcase_05 | AC | 17 ms
8,024 KB |
testcase_06 | AC | 136 ms
8,884 KB |
testcase_07 | AC | 413 ms
10,292 KB |
testcase_08 | AC | 174 ms
9,060 KB |
testcase_09 | AC | 463 ms
9,180 KB |
testcase_10 | AC | 287 ms
9,512 KB |
testcase_11 | AC | 1,152 ms
11,080 KB |
testcase_12 | AC | 20 ms
8,460 KB |
testcase_13 | AC | 273 ms
8,856 KB |
testcase_14 | AC | 336 ms
9,832 KB |
testcase_15 | AC | 191 ms
9,184 KB |
testcase_16 | AC | 40 ms
8,576 KB |
testcase_17 | AC | 137 ms
9,284 KB |
testcase_18 | AC | 533 ms
10,468 KB |
testcase_19 | AC | 214 ms
9,284 KB |
testcase_20 | AC | 106 ms
8,884 KB |
testcase_21 | AC | 52 ms
8,728 KB |
testcase_22 | AC | 197 ms
9,416 KB |
testcase_23 | AC | 17 ms
8,052 KB |
testcase_24 | AC | 16 ms
8,056 KB |
testcase_25 | AC | 16 ms
8,076 KB |
testcase_26 | AC | 114 ms
9,144 KB |
testcase_27 | AC | 33 ms
8,428 KB |
testcase_28 | AC | 68 ms
8,708 KB |
testcase_29 | AC | 113 ms
9,044 KB |
testcase_30 | AC | 399 ms
9,500 KB |
testcase_31 | AC | 1,095 ms
11,152 KB |
ソースコード
import sys sys.setrecursionlimit(10 ** 6) int1 = lambda x: int(x) - 1 p2D = lambda x: print(*x, sep="\n") def II(): return int(sys.stdin.buffer.readline()) def LI(): return list(map(int, sys.stdin.buffer.readline().split())) def LLI(rows_number): return [LI() for _ in range(rows_number)] def LI1(): return list(map(int1, sys.stdin.buffer.readline().split())) def LLI1(rows_number): return [LI1() for _ in range(rows_number)] def BI(): return sys.stdin.buffer.readline().rstrip() def SI(): return sys.stdin.buffer.readline().rstrip().decode() inf = 10 ** 16 md = 10 ** 9 + 7 # md = 998244353 n,k=LI() to=[[] for _ in range(n)] cc=[] for i in range(n-1): u,v,c=LI() u,v=u-1,v-1 to[u].append((v,c,i)) to[v].append((u,c,i)) cc.append(c) leaf=[0]*n etou=[0]*n def dfs(u=0,pu=-1,s=0): if u and len(to[u])==1: leaf[u]=1 return s res=0 for v,c,e in to[u]: if v==pu:continue etou[e]=v res+=dfs(v,u,s+c) leaf[u]+=leaf[v] return res score=dfs() dp=[-1]*(k+1) dp[0]=0 for e in range(n-1): c=cc[e] u=etou[e] for i in range(k-c,-1,-1): if dp[i]==-1:continue ni=i+c dp[ni]=max(dp[ni],dp[i]+c*leaf[u]) print(score+max(dp))