結果

問題 No.3139 Interval MEX ?
ユーザー 👑 potato167
提出日時 2025-05-01 22:34:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 354 ms / 2,000 ms
コード長 665 bytes
コンパイル時間 413 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 92,800 KB
最終ジャッジ日時 2025-05-02 19:36:23
合計ジャッジ時間 16,547 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 57
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353
N, M = map(int, input().split())
L = 1001001
fact = [1] * L
fact_inv = [1] * L
for i in range(L - 1):
	fact[i + 1] = fact[i] * (i + 1) % MOD
fact_inv[L - 1] = pow(fact[L - 1], -1, MOD)
for i in range(L - 1, 0, -1):
	fact_inv[i - 1] = fact_inv[i] * i % MOD
M -= 1
ans = fact[N + M] * fact_inv[M] % MOD
for x in range(1, N + 1):
	n_ans = fact[N - x + M] * fact_inv[M] % MOD
	if x <= M:
		n_ans = n_ans * fact[N] % MOD
		n_ans = n_ans * fact_inv[N - x] % MOD
	else:
		n_ans = n_ans * fact[N - x + M] % MOD
		n_ans = n_ans * fact_inv[N - x] % MOD
		n_ans = n_ans * pow(N + M - x + 1, x - M, MOD) % MOD
	print((ans - n_ans) % MOD)
	ans = n_ans
print(ans)
0