結果

問題 No.2356 Back Door Tour in Four Seasons
ユーザー miho-4
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0