結果

問題 No.1878 union-find の数え上げ
ユーザー lam6er
提出日時 2025-03-31 17:24:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 106 ms / 2,000 ms
コード長 398 bytes
コンパイル時間 249 ms
コンパイル使用メモリ 82,056 KB
実行使用メモリ 78,152 KB
最終ジャッジ日時 2025-03-31 17:24:40
合計ジャッジ時間 2,336 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

n = int(input())
if n == 1:
    print(1)
else:
    p = [0] * (n + 1)
    for i in range(2, n + 1):
        p[i] = int(input())
    
    depth = [0] * (n + 1)
    depth[1] = 0  # Root's depth is 0
    
    for u in range(2, n + 1):
        depth[u] = depth[p[u]] + 1
    
    result = 1
    for u in range(2, n + 1):
        result = (result * depth[u]) % MOD
    
    print(result)
0