結果
問題 | No.2582 Random Average^K |
ユーザー | ochiaigawa |
提出日時 | 2023-12-10 10:55:05 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 255 ms / 2,000 ms |
コード長 | 998 bytes |
コンパイル時間 | 2,177 ms |
コンパイル使用メモリ | 208,996 KB |
実行使用メモリ | 11,044 KB |
最終ジャッジ日時 | 2023-12-10 10:55:10 |
合計ジャッジ時間 | 4,689 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 15 ms
11,044 KB |
testcase_01 | AC | 14 ms
11,044 KB |
testcase_02 | AC | 14 ms
11,044 KB |
testcase_03 | AC | 14 ms
11,044 KB |
testcase_04 | AC | 14 ms
11,044 KB |
testcase_05 | AC | 14 ms
11,044 KB |
testcase_06 | AC | 15 ms
11,044 KB |
testcase_07 | AC | 15 ms
11,044 KB |
testcase_08 | AC | 17 ms
11,044 KB |
testcase_09 | AC | 79 ms
11,044 KB |
testcase_10 | AC | 119 ms
11,044 KB |
testcase_11 | AC | 120 ms
11,044 KB |
testcase_12 | AC | 255 ms
11,044 KB |
testcase_13 | AC | 240 ms
11,044 KB |
testcase_14 | AC | 15 ms
11,044 KB |
testcase_15 | AC | 254 ms
11,044 KB |
testcase_16 | AC | 139 ms
11,044 KB |
testcase_17 | AC | 142 ms
11,044 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/modint> #pragma GCC optimize("Ofast") #pragma GCC optimize("unroll-loops") using namespace std; using mint = atcoder::modint998244353; // for modint const int MAX = 1000005; vector<mint> fac(MAX, mint(1)), invfac(MAX, mint(1)); void COMinit() { for(int i = 2; i < MAX; i++){ fac[i] = fac[i - 1] * mint(i); } invfac.back() = mint(1) / fac.back(); for(int i = MAX - 1; i > 1; i--) { invfac[i - 1] = invfac[i] * i; } } mint COM(int n, int k){ if(n < k) return mint(0); if(n < 0 || k < 0) return mint(0); return fac[n] * invfac[k] * invfac[n - k]; } int main() { cin.tie(0); cout.tie(0); ios::sync_with_stdio(false); COMinit(); int N, K; cin >> N >> K; mint ans = 0; for(int i = 1; i <= N; i++) { ans += COM(N, i) * mint(i).pow(N + K) * (i % 2 == N % 2 ? 1 : -1); } for(int i = K + 1; i <= N + K; i++) { ans /= i; } ans /= mint(N).pow(K); cout << ans.val() << '\n'; return 0; }