結果
| 問題 |
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, random
from collections import deque, Counter, defaultdict
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
inf = 2 ** 63 - 1
mod = 998244353
class Combinatorics():
def __init__(self, mod, maxi):
self.mod = mod
self.maxi = maxi
self.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.mod
self.factinvs[i] = (self.factinvs[i-1] * self.invs[i]) % self.mod
def choose(self, n, k) -> int:
if k < 0 or k > n: return 0
if k == 0 or k == n: return 1
k = min(k, n - k)
return (((self.facs[n] * self.factinvs[k]) % self.mod) * self.factinvs[n-k]) % self.mod
def perm(self, n, k) -> int:
return (self.choose(n, k) * self.facs[k]) % self.mod
def homop(self, n, k) -> int:
if n == k == 0:
return 1
return self.choose(n + k - 1, k)
x, y, z, w = mi()
if z == 0:
w, z = z, w
x, y = y, x
counth = x - z
countw = y - 1
C = Combinatorics(mod, 10 ** 6)
print(C.choose(counth + countw, countw) * C.facs[y] % mod * C.perm(x, counth) % mod)