結果
問題 | 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,bisectfrom collections import deque,defaultdictfrom heapq import heapify,heappop,heappushfrom itertools import permutationsfrom math import gcdinput = lambda :sys.stdin.readline().rstrip()mi = lambda :map(int,input().split())li = lambda :list(mi())mod = 998244353N = 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] * Ndeq = deque([0])while deq:v = deq.popleft()for nv in edge[v]:depth[nv] = depth[v] + 1deq.append(nv)res = 1for v in range(1,N):res *= depth[v]res %= modprint(res)