結果

問題 No.417 チューリップバブル
ユーザー kept1994
提出日時 2022-10-24 21:21:16
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,420 bytes
コンパイル時間 1,485 ms
コンパイル使用メモリ 82,188 KB
実行使用メモリ 85,656 KB
最終ジャッジ日時 2024-07-03 00:17:39
合計ジャッジ時間 26,960 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
import sys

def main():
    N, M = map(int, input().split())
    G = [[] for _ in range(N)]
    U = [int(input()) for _ in range(N)]
    dp = [[U[i]] * (M + 1) for i in range(N)] # dp[node][i] := nodeにおいて、残り時間restで得られる最大のスコア
    for _ in range(N - 1):
        aa, bb, cc = map(int, input().split())
        G[aa].append((bb, cc))
        G[bb].append((aa, cc))

    def dfs(pre: int, now: int):
        for next, cost in G[now]: 
            if next == pre: 
                continue
            dfs(now, next)
            for rest in range(M, cost * 2 - 1, -1):
                for j in range(rest - cost * 2 + 1): 
                    dp[now][rest] = max(dp[now][rest], dp[now][rest - (j + cost * 2)] + dp[next][j])
                    # 子ノードnextに行くには(j + cost)だけ必要になる。
                    # -> その分を確保し、引いた後に残る時間でノードnowにおいて取れる最大のスコアdp[now][rest - (j + cost)] ( <- 他の子ノードを回る分があるのでこれを考慮する)
                    # -> と、子ノードnextに行ったことで得られるスコア dp[next][j] で更新する。
        return
    dfs(-1, 0)
    ans = -1
    for i in range(M + 1):
        # print(dp[0][i]) 
        ans = max(ans, dp[0][i]) 
    print(ans)    
    return


if __name__ == '__main__':
    main()
0