結果
問題 | No.2409 Strange Werewolves |
ユーザー |
|
提出日時 | 2023-08-11 21:32:35 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 153 ms / 2,000 ms |
コード長 | 1,408 bytes |
コンパイル時間 | 341 ms |
コンパイル使用メモリ | 82,248 KB |
実行使用メモリ | 87,232 KB |
最終ジャッジ日時 | 2024-11-18 15:30:40 |
合計ジャッジ時間 | 3,590 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 16 |
ソースコード
import sys, time, randomfrom collections import deque, Counter, defaultdictinput = lambda: sys.stdin.readline().rstrip()ii = lambda: int(input())mi = lambda: map(int, input().split())li = lambda: list(mi())inf = 2 ** 63 - 1mod = 998244353class Combinatorics():def __init__(self, mod, maxi):self.mod = modself.maxi = maxiself.facs = [1] * (maxi + 1)self.factinvs = [1] * (maxi + 1)self.invs = [1] * (maxi + 1)for i in range(2, self.maxi + 1):self.facs[i] = ((self.facs[i-1] * i) % self.mod)self.invs[i] = (-self.invs[self.mod % i] * (self.mod // i)) % self.modself.factinvs[i] = (self.factinvs[i-1] * self.invs[i]) % self.moddef choose(self, n, k) -> int:if k < 0 or k > n: return 0if k == 0 or k == n: return 1k = min(k, n - k)return (((self.facs[n] * self.factinvs[k]) % self.mod) * self.factinvs[n-k]) % self.moddef perm(self, n, k) -> int:return (self.choose(n, k) * self.facs[k]) % self.moddef homop(self, n, k) -> int:if n == k == 0:return 1return self.choose(n + k - 1, k)x, y, z, w = mi()if z == 0:w, z = z, wx, y = y, xcounth = x - zcountw = y - 1C = Combinatorics(mod, 10 ** 6)print(C.choose(counth + countw, countw) * C.facs[y] % mod * C.perm(x, counth) % mod)