結果
| 問題 |
No.3283 Labyrinth and Friends
|
| コンテスト | |
| ユーザー |
detteiuu
|
| 提出日時 | 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:]))
detteiuu