結果

問題 No.1488 Max Score of the Tree
ユーザー H20
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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))
0