結果

問題 No.2409 Strange Werewolves
ユーザー detteiuu
提出日時 2024-12-22 05:26:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 856 ms / 2,000 ms
コード長 854 bytes
コンパイル時間 333 ms
コンパイル使用メモリ 82,296 KB
実行使用メモリ 228,500 KB
最終ジャッジ日時 2024-12-22 05:27:09
合計ジャッジ時間 17,090 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

class CP:
    def __init__(self, N):
        self.fact = [1]
        self.fact_inv = [1]
        for i in range(1, N+1):
            self.fact.append((self.fact[-1]*i)%MOD)
            self.fact_inv.append(pow(self.fact[-1], -1, MOD))
    def C(self, N, K):
        if N >= K:
            return self.fact[N]*self.fact_inv[K]%MOD*self.fact_inv[N-K]%MOD
        else:
            return 0
    def P(self, N, K):
        if N >= K:
            return self.fact[N]*self.fact_inv[N-K]%MOD
        else:
            return 0

X, Y, Z, W = map(int, input().split())

MOD = 998244353

if W != 0:
    X, Y = Y, X
    Z, W = W, Z

if Y == 0:
    exit(print(1))

P = [1]
for i in range(1, 10**6+1):
    P.append(P[-1]*i%MOD)

diffX = X-Z
cp = CP(10**6)
ans = cp.C(diffX+Y-1, diffX)
ans *= cp.C(X, diffX)
ans %= MOD
ans *= (P[diffX]*P[Y])%MOD
ans %= MOD

print(ans)
0