結果

問題 No.3217 Shiki no Shiki
コンテスト
ユーザー moshi777
提出日時 2025-08-01 21:41:17
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 585 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 82,284 KB
実行使用メモリ 98,716 KB
最終ジャッジ日時 2025-08-01 21:41:25
合計ジャッジ時間 4,680 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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)]
g_rev = [list() for i in range(N)]
for i in range(N):
    if P[i] != 0:
        g[P[i] - 1].append(i)
        g_rev[i].append(P[i]-1)

depth = [-1] * N


ans = 0


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)

is_ss = [False] * N

ans = 0
for i in range(N):
    if depth[i] >= 2:
        j = P[P[i] - 1]-1
        is_ss[j] = True
ans = sum(is_ss)
print(ans)
0