結果

問題 No.2409 Strange Werewolves
ユーザー FromBooskaFromBooska
提出日時 2023-08-12 04:59:26
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 669 ms / 2,000 ms
コード長 1,009 bytes
コンパイル時間 166 ms
コンパイル使用メモリ 82,036 KB
実行使用メモリ 146,304 KB
最終ジャッジ日時 2024-04-29 19:31:35
合計ジャッジ時間 7,185 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,352 KB
testcase_01 AC 47 ms
59,776 KB
testcase_02 AC 39 ms
51,840 KB
testcase_03 AC 663 ms
145,920 KB
testcase_04 AC 665 ms
146,176 KB
testcase_05 AC 358 ms
102,528 KB
testcase_06 AC 669 ms
146,304 KB
testcase_07 AC 669 ms
146,176 KB
testcase_08 AC 384 ms
107,904 KB
testcase_09 AC 427 ms
113,280 KB
testcase_10 AC 236 ms
86,016 KB
testcase_11 AC 87 ms
64,512 KB
testcase_12 AC 235 ms
86,400 KB
testcase_13 AC 388 ms
107,392 KB
testcase_14 AC 443 ms
113,408 KB
testcase_15 AC 410 ms
112,896 KB
testcase_16 AC 342 ms
102,144 KB
testcase_17 AC 83 ms
65,280 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