結果
問題 | 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;}