結果
問題 | No.2381 Gift Exchange Party |
ユーザー |
|
提出日時 | 2023-07-14 21:37:18 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 61 ms / 2,000 ms |
コード長 | 845 bytes |
コンパイル時間 | 2,943 ms |
コンパイル使用メモリ | 246,240 KB |
実行使用メモリ | 6,400 KB |
最終ジャッジ日時 | 2024-09-16 06:29:26 |
合計ジャッジ時間 | 4,191 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
#include <bits/stdc++.h>using namespace std;using i64 = int64_t;constexpr i64 mod = 998244353;i64 power(i64 a, i64 r) {i64 res = 1;for (; r; r >>= 1, a = a * a % mod) {if (r & 1) {res = res * a % mod;}}return res;}int main() {ios::sync_with_stdio(false);cin.tie(nullptr);i64 n, p;cin >> n >> p;vector<i64> f(n + 1, 1), g(n + 1, 1);for (i64 i = 1; i <= n; i += 1) {g[i] = power(f[i] = f[i - 1] * i % mod, mod - 2);}i64 ans = f[n];for (i64 i = 0; i * p <= n; i += 1) {i64 pans = f[n];pans = pans * g[n - i * p] % mod;pans = pans * power(power(p, mod - 2), i) % mod;pans = pans * g[i] % mod;// cerr << i << " " << pans << "\n";ans = (ans + mod - pans) % mod;}cout << ans;}