結果

問題 No.3217 Shiki no Shiki
ユーザー moshi777
提出日時 2025-08-01 21:35:45
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 467 bytes
コンパイル時間 249 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 98,692 KB
最終ジャッジ日時 2025-08-01 21:35:50
合計ジャッジ時間 4,678 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 3 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
P = list(map(int, input().split()))

g = [list() for i in range(N)]
for i in range(N):
    if P[i] != 0:
        g[P[i] - 1].append(i)

depth = [-1] * N


def dfs(v, d=0, pre=-1):
    depth[v] = d
    for u in g[v]:
        if u == pre:
            continue
        dfs(u, d + 1, pre=v)


for i in range(N):
    if P[i] == 0:
        dfs(i)

ans = 0
for i in range(N):
    ok = (depth[i] >= 2 and len(g[i]) == 0)
    ans += 1 if ok else 0
print(ans)
0