結果

問題 No.2530 Yellow Cards
ユーザー cho435
提出日時 2023-11-03 23:11:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,705 ms / 2,000 ms
コード長 516 bytes
コンパイル時間 4,194 ms
コンパイル使用メモリ 252,640 KB
最終ジャッジ日時 2025-02-17 18:50:24
ジャッジサーバーID
(参考情報)
judge5 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i,n) for(int i=0;i<(int)(n);i++)
using mint = atcoder::modint998244353;

int main(){
	int n,k;
	cin>>n>>k;
	vector<mint> dp(n+1,0);
	dp.at(0)=1;
	rep(i,k){
		vector<mint> ndp(n+1,0);
		rep(j,n+1){
			if(dp.at(j)==0) continue;
			if(j<n) ndp.at(j+1)+=(dp.at(j))*(n-j)/n;
			if(j>0) ndp.at(j-1)+=(dp.at(j))*(j)/n;
		}
		swap(dp,ndp);
	}
	mint ans=0;
	rep(i,n+1) ans+=dp.at(i)*(n+(k-i)/2);
	cout<<ans.val()<<endl;
}
0