結果
問題 | No.1488 Max Score of the Tree |
ユーザー |
![]() |
提出日時 | 2021-04-23 23:01:53 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 110 ms / 2,000 ms |
コード長 | 754 bytes |
コンパイル時間 | 155 ms |
コンパイル使用メモリ | 81,964 KB |
実行使用メモリ | 76,540 KB |
最終ジャッジ日時 | 2024-07-04 08:38:30 |
合計ジャッジ時間 | 3,267 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
N,K = map(int, input().split()) L = [0]*(N-1) C = [0]*(N-1) KI = [] d = {} for i in range(N): KI.append([]) for i in range(N-1): a,b,c=map(int, input().split()) KI[a-1].append(b-1) KI[b-1].append(a-1) d[(a-1,b-1)]=i d[(b-1,a-1)]=i C[i]=c def dfs(p,pre,no): if len(KI[p])==1 and p!=0: for n in no: L[d[n]]+=1 else: for ki in KI[p]: if pre==ki: continue dfs(ki,p,no+[(p,ki)]) dfs(0,-1,[]) ans = 0 for i in range(N-1): ans+=L[i]*C[i] DP = [-1]*(K+1) DP[0] = 0 for i in range(N-1): for j in reversed(range(max(0,K+1-C[i]))): if DP[j]>=0 and DP[j+C[i]]<DP[j]+C[i]*L[i]: DP[j+C[i]] = DP[j]+C[i]*L[i] print(ans+max(DP))