結果

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

ソースコード

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):
    ans += 1 if depth[i] == 2 else 0
print(ans)
0