結果

問題 No.417 チューリップバブル
ユーザー Yukino DX.
提出日時 2023-11-11 12:17:05
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 696 bytes
コンパイル時間 297 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 17,824 KB
最終ジャッジ日時 2024-09-26 02:42:43
合計ジャッジ時間 4,449 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 8 TLE * 2 -- * 30
権限があれば一括ダウンロードができます

ソースコード

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