結果

問題 No.2409 Strange Werewolves
ユーザー FromBooska
提出日時 2023-08-12 04:59:26
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 615 ms / 2,000 ms
コード長 1,009 bytes
コンパイル時間 297 ms
コンパイル使用メモリ 82,424 KB
実行使用メモリ 146,560 KB
最終ジャッジ日時 2024-11-19 02:18:21
合計ジャッジ時間 6,628 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

# dpかと思ったが順列組合せだろう
# dpには制約数字が大きすぎるし
# 入力例1で考える
# 最後の狼が誰かでY通り、残る人間が誰かでxCz通り、
# それまでに選ばれるのが(Y-1)+(X-Z)人でその全並びなのでファクトリアル

X, Y, Z, W = map(int, input().split())
N = max(X, Y, X+Y)
mod = 998244353

# nCrメモ化パッケージ
factorial = [1] #0分
inverse = [1] #0分
for i in range(1, N+1):
    factorial.append(factorial[-1]*i%mod)
    inverse.append(pow(factorial[-1], mod-2, mod))
    
def nCr_fast(N, R, MOD):
    if N < R or R < 0:
        return 0
    elif R == 0 or R == N:
        return 1
    return factorial[N]*inverse[R]*inverse[N-R]%MOD

if W == 0:
    ans = 1
    ans *= Y
    ans *= nCr_fast(X, Z, mod)
    ans *= factorial[(Y-1)+(X-Z)]
    ans %= mod
    print(ans)
elif Z == 0:
    ans = 1
    ans *= X
    ans *= nCr_fast(Y, W, mod)
    ans *= factorial[(X-1)+(Y-W)]
    ans %= mod
    print(ans)    

#print(factorial[:5])
0