結果
問題 | No.1960 Guruguru Permutation |
ユーザー | 👑 rin204 |
提出日時 | 2022-05-27 21:55:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 63 ms / 2,000 ms |
コード長 | 924 bytes |
コンパイル時間 | 355 ms |
コンパイル使用メモリ | 82,352 KB |
実行使用メモリ | 67,200 KB |
最終ジャッジ日時 | 2024-09-20 15:48:13 |
合計ジャッジ時間 | 2,324 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
MOD = 998244353 N = 200200 fact = [0 for _ in range(N)] invfact = [0 for _ in range(N)] fact[0] = 1 for i in range(1, N): fact[i] = fact[i - 1] * i % MOD invfact[N - 1] = pow(fact[N - 1], MOD - 2, MOD) for i in range(N - 2, -1, -1): invfact[i] = invfact[i + 1] * (i + 1) % MOD def nCk(n, k): if k < 0 or n < k: return 0 else: return (fact[n] * invfact[k] % MOD) * invfact[n - k] % MOD def nHk(n, k): return nCk(n + k - 1, k) n, m, k = map(int, input().split()) if m + k >= n: x = m + k - n m -= x k -= x ans = 0 for i in range(min(m, k) + 1): ans += (nCk(m, i) * nCk(k, i) % MOD) * fact[i] % MOD ans %= MOD print(ans) else: ans = 0 for i in range(min(m, k) + 1): ans += (nCk(m, i) * nCk(k, i) % MOD) * fact[i] % MOD ans %= MOD for i in range(m + k + 1, n + 1): ans *= i ans %= MOD print(ans)