結果
問題 | No.2530 Yellow Cards |
ユーザー |
![]() |
提出日時 | 2023-11-03 22:00:17 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 451 ms / 2,000 ms |
コード長 | 805 bytes |
コンパイル時間 | 2,260 ms |
コンパイル使用メモリ | 199,012 KB |
最終ジャッジ日時 | 2025-02-17 18:15:19 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 998244353; int main(){ int N, K; cin >> N >> K; vector<long long> inv(N + 1); inv[1] = 1; for (int i = 2; i <= N; i++){ inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD; } vector<vector<long long>> dp(K + 1, vector<long long>(N + K + 1, 0)); dp[0][0] = 1; for (int i = 0; i < K; i++){ for (int j = 0; j < N + K; j++){ if (dp[i][j] > 0){ int c = i - j * 2; dp[i + 1][j] += dp[i][j] * (N - c) % MOD * inv[N] % MOD; dp[i + 1][j] %= MOD; dp[i + 1][j + 1] += dp[i][j] * c % MOD * inv[N] % MOD; dp[i + 1][j + 1] %= MOD; } } } long long ans = 0; for (int i = 0; i <= N + K; i++){ ans += dp[K][i] * (N + i) % MOD; } ans %= MOD; cout << ans << endl; }