結果
| 問題 |
No.2033 Chromatic Duel
|
| コンテスト | |
| ユーザー |
gew1fw
|
| 提出日時 | 2025-06-12 17:00:55 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 872 bytes |
| コンパイル時間 | 199 ms |
| コンパイル使用メモリ | 82,064 KB |
| 実行使用メモリ | 63,056 KB |
| 最終ジャッジ日時 | 2025-06-12 17:01:10 |
| 合計ジャッジ時間 | 2,983 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 WA * 3 |
| other | AC * 6 WA * 31 |
ソースコード
MOD = 998244353
def main():
import sys
input = sys.stdin.read().split()
N = int(input[0])
B = int(input[1])
W = int(input[2])
if B < 0 or W < 0 or (B + W) > N:
print(0)
return
max_n = B + W
if max_n == 0:
print(1 if B == 0 and W == 0 else 0)
return
# 预计算阶乘和逆阶乘
fact = [1] * (max_n + 1)
for i in range(1, max_n + 1):
fact[i] = fact[i-1] * i % MOD
inv_fact = [1] * (max_n + 1)
inv_fact[max_n] = pow(fact[max_n], MOD-2, MOD)
for i in range(max_n-1, -1, -1):
inv_fact[i] = inv_fact[i+1] * (i+1) % MOD
def comb(n, k):
if n < 0 or k < 0 or k > n:
return 0
return fact[n] * inv_fact[k] % MOD * inv_fact[n - k] % MOD
res = comb(B + W, B)
print(res)
if __name__ == "__main__":
main()
gew1fw