結果

問題 No.2584 The University of Tree
ユーザー hirayuu_ychirayuu_yc
提出日時 2023-11-27 21:22:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,738 ms / 3,000 ms
コード長 1,868 bytes
コンパイル時間 190 ms
コンパイル使用メモリ 81,572 KB
実行使用メモリ 261,612 KB
最終ジャッジ日時 2023-12-11 23:31:07
合計ジャッジ時間 40,487 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
55,596 KB
testcase_01 AC 38 ms
55,596 KB
testcase_02 AC 38 ms
55,596 KB
testcase_03 AC 37 ms
55,596 KB
testcase_04 AC 49 ms
63,320 KB
testcase_05 AC 49 ms
63,312 KB
testcase_06 AC 49 ms
63,324 KB
testcase_07 AC 49 ms
63,320 KB
testcase_08 AC 56 ms
63,316 KB
testcase_09 AC 50 ms
63,316 KB
testcase_10 AC 50 ms
63,316 KB
testcase_11 AC 58 ms
63,316 KB
testcase_12 AC 50 ms
63,316 KB
testcase_13 AC 50 ms
63,316 KB
testcase_14 AC 49 ms
63,320 KB
testcase_15 AC 49 ms
62,820 KB
testcase_16 AC 50 ms
63,324 KB
testcase_17 AC 50 ms
63,320 KB
testcase_18 AC 50 ms
63,316 KB
testcase_19 AC 50 ms
63,316 KB
testcase_20 AC 49 ms
63,324 KB
testcase_21 AC 51 ms
63,320 KB
testcase_22 AC 879 ms
154,036 KB
testcase_23 AC 967 ms
196,092 KB
testcase_24 AC 659 ms
127,460 KB
testcase_25 AC 748 ms
134,832 KB
testcase_26 AC 533 ms
122,344 KB
testcase_27 AC 879 ms
158,792 KB
testcase_28 AC 526 ms
115,652 KB
testcase_29 AC 384 ms
98,784 KB
testcase_30 AC 1,703 ms
221,428 KB
testcase_31 AC 1,641 ms
226,220 KB
testcase_32 AC 459 ms
109,536 KB
testcase_33 AC 769 ms
226,116 KB
testcase_34 AC 377 ms
114,904 KB
testcase_35 AC 1,642 ms
223,368 KB
testcase_36 AC 1,548 ms
254,852 KB
testcase_37 AC 1,573 ms
208,704 KB
testcase_38 AC 1,571 ms
209,280 KB
testcase_39 AC 1,618 ms
244,780 KB
testcase_40 AC 1,664 ms
241,332 KB
testcase_41 AC 1,638 ms
226,744 KB
testcase_42 AC 1,738 ms
222,676 KB
testcase_43 AC 1,713 ms
221,940 KB
testcase_44 AC 1,661 ms
222,272 KB
testcase_45 AC 1,681 ms
221,876 KB
testcase_46 AC 786 ms
227,100 KB
testcase_47 AC 1,424 ms
261,612 KB
testcase_48 AC 678 ms
128,468 KB
testcase_49 AC 1,423 ms
197,624 KB
testcase_50 AC 1,229 ms
177,144 KB
testcase_51 AC 728 ms
134,600 KB
testcase_52 AC 419 ms
104,452 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
MOD=998244353
N=int(input())
gr=[]
road=[[] for i in range(N-1)]
for i in range(N):
    C,*E=list(map(int,input().split()))
    gr.append(E)
    for j in E:
        road[j-1].append(i)
ser=[{} for i in range(N)]
for i in range(N):
    for j,k in enumerate(gr[i]):
        if road[k-1][0]==i:
            gr[i][j]=road[k-1][1]
        else:
            gr[i][j]=road[k-1][0]
        ser[i][gr[i][j]]=j
dp=[[-1]*len(gr[i]) for i in range(N)]
top=[]
vert=deque([(0,0,-1)])
while len(vert)>0:
    st,pos,bef=vert.pop()
    if st==0:
        top.append(pos)
        vert.append((1,pos,bef))
        for i in gr[pos]:
            if i!=bef:
                vert.append((0,i,pos))
    else:
        if bef==-1:
            continue
        prod=pos+1
        ret=0
        for i in range(ser[pos][bef]+1,len(gr[pos])):
            ret+=dp[pos][ser[pos][gr[pos][i]]]*prod
            prod*=pos+1
            ret%=MOD
            prod%=MOD
        ret+=prod
        prod*=pos+1
        for i in range(ser[pos][bef]):
            ret+=dp[pos][ser[pos][gr[pos][i]]]*prod
            prod*=pos+1
            ret%=MOD
            prod%=MOD
        dp[bef][ser[bef][pos]]=ret
for i in top:
    lf=[0]
    ri=[0]
    rev=pow(i+1,-1,MOD)
    prod=i+1
    for j in dp[i]:
        lf.append(lf[-1]+j*prod)
        prod*=i+1
        prod%=MOD
        lf[-1]%=MOD
    prod=rev
    for j in reversed(dp[i]):
        ri.append(ri[-1]+j*prod)
        prod*=rev
        prod%=MOD
        ri[-1]%=MOD
    cul=-2
    cur=0
    prod=i+1
    for j in reversed(gr[i]):
        dp[j][ser[j][i]]=prod+(lf[cul]+ri[cur])*prod
        dp[j][ser[j][i]]%=MOD
        prod*=i+1
        prod%=MOD
        cur+=1
        cul-=1
for i in range(N):
    ans=0
    prod=i+1
    for j in dp[i]:
        ans+=j*prod
        ans%=MOD
        prod*=i+1
        prod%=MOD
    print(ans)
0