結果

問題 No.2409 Strange Werewolves
ユーザー FromBooskaFromBooska
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 34 ms
51,584 KB
testcase_01 AC 44 ms
60,032 KB
testcase_02 AC 36 ms
51,968 KB
testcase_03 AC 615 ms
145,920 KB
testcase_04 AC 606 ms
146,560 KB
testcase_05 AC 325 ms
102,272 KB
testcase_06 AC 605 ms
146,176 KB
testcase_07 AC 613 ms
145,920 KB
testcase_08 AC 352 ms
107,520 KB
testcase_09 AC 379 ms
113,408 KB
testcase_10 AC 201 ms
85,888 KB
testcase_11 AC 70 ms
64,896 KB
testcase_12 AC 204 ms
86,272 KB
testcase_13 AC 344 ms
107,520 KB
testcase_14 AC 406 ms
114,048 KB
testcase_15 AC 377 ms
113,408 KB
testcase_16 AC 310 ms
102,272 KB
testcase_17 AC 76 ms
65,536 KB
権限があれば一括ダウンロードができます

ソースコード

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