結果
問題 | No.1878 union-find の数え上げ |
ユーザー |
|
提出日時 | 2022-03-18 22:06:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 129 ms / 2,000 ms |
コード長 | 667 bytes |
コンパイル時間 | 396 ms |
コンパイル使用メモリ | 82,980 KB |
実行使用メモリ | 90,052 KB |
最終ジャッジ日時 | 2024-10-03 09:40:11 |
合計ジャッジ時間 | 2,313 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 13 |
ソースコード
import sys,random,bisect from collections import deque,defaultdict from heapq import heapify,heappop,heappush from itertools import permutations from math import gcd input = lambda :sys.stdin.readline().rstrip() mi = lambda :map(int,input().split()) li = lambda :list(mi()) mod = 998244353 N = int(input()) parent = [-1] + [int(input())-1 for i in range(N-1)] edge = [[] for v in range(N)] for i in range(1,N): edge[parent[i]].append(i) depth = [0] * N deq = deque([0]) while deq: v = deq.popleft() for nv in edge[v]: depth[nv] = depth[v] + 1 deq.append(nv) res = 1 for v in range(1,N): res *= depth[v] res %= mod print(res)