結果

問題 No.3449 Mex of Subtree
コンテスト
ユーザー kidodesu
提出日時 2026-02-20 22:02:10
言語 PyPy3
(7.3.17)
結果
AC  
実行時間 733 ms / 2,000 ms
コード長 831 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 345 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 166,884 KB
最終ジャッジ日時 2026-02-20 22:02:22
合計ジャッジ時間 11,417 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 59
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

n = int(input())
P = [-1] + list(map(lambda x: int(x)-1, input().split()))
node = [[] for _ in range(n)]
for i in range(1, n):
    node[P[i]].append(i)

mod = 998244353
E = []
D = [-1] * n
D[0] = 0
A = [0]
while A:
    now = A.pop()
    E.append(now)
    for nxt in node[now]:
        if D[nxt] == -1:
            D[nxt] = D[now] + 1
            A.append(nxt)
dp = [[0, 1] for _ in range(n)]
for now in E[::-1]:
    l = len(dp[now])
    for i in range(l-2, -1, -1):
        dp[now][i] = (dp[now][i] + dp[now][i+1]) % mod
    nxt = P[now]
    if nxt == -1:
        continue
    a = len(dp[now])
    b = len(dp[nxt])
    c = a + b - 1
    ndp = [0] * c
    for ai in range(a):
        for bi in range(b):
            ndp[ai+bi] = (ndp[ai+bi] + dp[now][ai] * dp[nxt][bi]) % mod
    dp[nxt] = ndp

#print(dp[0])
print(sum(dp[0]) % mod)
0