結果

問題 No.417 チューリップバブル
ユーザー Yukino DX.
提出日時 2023-11-11 12:16:22
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 696 bytes
コンパイル時間 226 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 82,688 KB
最終ジャッジ日時 2024-09-26 02:42:35
合計ジャッジ時間 13,695 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 20 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

from itertools import count

n, m = map(int, input().split())
u = [int(input()) for _ in range(n)]
g = [[] for _ in range(n)]
for _ in range(n - 1):
    a, b, c = map(int, input().split())
    g[a].append((b, c))
    g[b].append((a, c))


dp = [[0] * (m + 1) for _ in range(n)]


def dfs(crr, prv):
    for nxt, _ in g[crr]:
        if nxt != prv:
            dfs(nxt, crr)

    for nxt, t in g[crr]:
        for i in range(m, -1, -1):
            for j in count(0):
                if i < 2 * t + j:
                    break
                dp[crr][i] = max(dp[crr][i], dp[crr][i - (2 * t + j)] + dp[nxt][j])

    for i in range(m + 1):
        dp[crr][i] += u[crr]


dfs(0, n)
print(dp[0][m])
0