結果
問題 | No.2530 Yellow Cards |
ユーザー |
![]() |
提出日時 | 2023-11-03 22:30:28 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,807 ms / 2,000 ms |
コード長 | 717 bytes |
コンパイル時間 | 2,109 ms |
コンパイル使用メモリ | 196,880 KB |
最終ジャッジ日時 | 2025-02-17 18:34:27 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 17 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define int long longint mod = 998244353;long long modinv(long long a) {long long b = mod, u = 1, v = 0;while (b) {long long t = a / b;a -= t * b; swap(a, b);u -= t * v; swap(u, v);}u %= mod;if (u < 0) u += mod;return u;}signed main(){int N,K;cin>>N>>K;int k = modinv(N);vector<int> dp(N+1,0);dp[0] = 1;for(int i=0;i<K;i++){vector<int> ddp(N+1,0);for(int j=0;j<N;j++) ddp[j+1] = (ddp[j+1]+dp[j]*k%mod*(N-j)%mod)%mod;for(int j=N;j>0;j--) ddp[j-1] = (ddp[j-1]+dp[j]*k%mod*j%mod)%mod;dp = ddp;}int ans = 0;for(int i=0;i<=N;i++) ans = (ans+(N+(K-i)/2)*dp[i])%mod;cout<<ans<<endl;}