結果
問題 | No.2356 Back Door Tour in Four Seasons |
ユーザー |
|
提出日時 | 2023-06-16 22:39:39 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 726 bytes |
コンパイル時間 | 170 ms |
コンパイル使用メモリ | 82,028 KB |
実行使用メモリ | 144,048 KB |
最終ジャッジ日時 | 2024-06-24 15:23:04 |
合計ジャッジ時間 | 4,346 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 WA * 1 |
other | TLE * 1 -- * 27 |
ソースコード
mod=998244353 n=int(input()) L=[list(input().split()) for _ in range(n)] E=[[] for _ in range(4)] d={"U":0,"F":1,"W":2,"P":3} for a,b in L: E[d[a]].append(int(b)) e1,e2,e3,e4=len(E[0]),len(E[1]),len(E[2]),len(E[3]) dp2,dp3,dp4=[0]*e2,[0]*e3,[0]*e4 for i in range(e1): for j in range(e2): x=E[0][i] dp2[j]+=(pow(n-1,x,mod)-pow(n-2,x,mod))%mod dp2[j]%=mod for i in range(e2): for j in range(e3): x=E[1][i] dp3[j]+=dp2[i]*(pow(n-1,x,mod)-pow(n-2,x,mod))%mod dp3[j]%=mod for i in range(e3): for j in range(e4): x=E[2][i] dp4[j]+=dp3[i]*(pow(n-1,x,mod)-pow(n-2,x,mod))%mod dp4[j]%=mod ans=0 for i in range(e4): ans+=dp4[i]*pow(n-1,E[3][i],mod)%mod ans%=mod print(ans)