結果
問題 |
No.2530 Yellow Cards
|
ユーザー |
|
提出日時 | 2025-07-04 23:58:16 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 349 ms / 2,000 ms |
コード長 | 564 bytes |
コンパイル時間 | 4,898 ms |
コンパイル使用メモリ | 255,140 KB |
実行使用メモリ | 101,120 KB |
最終ジャッジ日時 | 2025-07-04 23:58:26 |
合計ジャッジ時間 | 7,742 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <atcoder/all> using namespace atcoder; using mint = modint998244353; int main () { int N, K; cin >> N >> K; std::vector dp(K + 1, vector(N + 1, mint(0))); mint iv = mint(N).inv(); dp[0][0] = 1; for (int k = 0; k < K; k ++) { for (int n = 0; n <= N; n ++) { if (n) { dp[k+1][n-1] += dp[k][n] * n * iv; } if (n < N) { dp[k+1][n+1] += dp[k][n] * (N - n) * iv; } } } mint ans = N; for (int n = 0; n <= N; n ++) { ans += dp[K][n] * ((K - n) / 2); } cout << ans.val() << endl; }