結果

問題 No.2409 Strange Werewolves
ユーザー timi
提出日時 2023-08-12 12:01:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 239 ms / 2,000 ms
コード長 452 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 82,220 KB
実行使用メモリ 252,216 KB
最終ジャッジ日時 2024-11-19 13:02:25
合計ジャッジ時間 5,378 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

x,y,z,w=map(int, input().split())
mod=998244353;n=10**6+100
inv_t=[0]+[1]
for i in range(2,n):
  inv_t+=[inv_t[mod%i]*(mod-int(mod/i))%mod]

kai=[1,1]
rev_kai=[1,inv_t[1]]
for i in range(2,n):
  kai.append(kai[-1]*i%mod)
  rev_kai.append(rev_kai[-1]*inv_t[i]%mod)

def cmb(n,r):
  return kai[n]*rev_kai[r]*rev_kai[n-r]%mod

if z==0:
  ans=x*cmb(y,w)
  z+=1
else:
  ans=y*cmb(x,z)
  w+=1

c=x+y-z-w
for i in range(1,c+1):
  ans*=i
  ans%=mod 
print(ans)
0