結果
問題 |
No.3283 Labyrinth and Friends
|
ユーザー |
![]() |
提出日時 | 2025-09-26 22:25:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 152 ms / 2,000 ms |
コード長 | 1,261 bytes |
コンパイル時間 | 287 ms |
コンパイル使用メモリ | 82,660 KB |
実行使用メモリ | 78,432 KB |
最終ジャッジ日時 | 2025-09-26 22:25:12 |
合計ジャッジ時間 | 5,146 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 45 |
ソースコード
from types import GeneratorType def bootstrap(f, stack=[]): def wrappedfunc(*args, **kwargs): if stack: return f(*args, **kwargs) to = f(*args, **kwargs) while True: if type(to) is GeneratorType: stack.append(to) to = next(to) else: stack.pop() if not stack: break to = stack[-1].send(to) return to return wrappedfunc INF = 1<<60 N, X = map(int, input().split()) P = list(map(int, input().split())) CS = [[0, 0]]+[list(map(int, input().split())) for _ in range(N-1)] C, S = map(list, zip(*CS)) G = [[] for _ in range(N)] for i, p in enumerate(P): G[p-1].append(i+1) @bootstrap def dfs(n, p): size[n] = S[n] dp[n] = [INF]*(size[n]+1) dp[n][-1] = C[n] for v in G[n]: if v == p: continue yield dfs(v, n) merge = [INF]*(size[n]+size[v]+1) for i in range(size[n]+1): for j in range(size[v]+1): merge[i+j] = min(merge[i+j], dp[n][i]+dp[v][j]) size[n] += size[v] dp[n] = merge dp[n][0] = 0 yield dp = [[] for _ in range(N)] size = [0]*N dfs(0, -1) print(min(dp[0][X:]))