結果
問題 |
No.2530 Yellow Cards
|
ユーザー |
![]() |
提出日時 | 2023-11-03 21:51:25 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 95 ms / 2,000 ms |
コード長 | 1,312 bytes |
コンパイル時間 | 3,632 ms |
コンパイル使用メモリ | 257,476 KB |
最終ジャッジ日時 | 2025-02-17 18:07:18 |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 17 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf32 1000000001 #define Inf64 1000000000000000001 struct combi{ deque<mint> kaijou; deque<mint> kaijou_; combi(int n){ kaijou.push_back(1); for(int i=1;i<=n;i++){ kaijou.push_back(kaijou[i-1]*i); } mint b=kaijou[n].inv(); kaijou_.push_front(b); for(int i=1;i<=n;i++){ int k=n+1-i; kaijou_.push_front(kaijou_[0]*k); } } mint combination(int n,int r){ if(r>n)return 0; mint a = kaijou[n]*kaijou_[r]; a *= kaijou_[n-r]; return a; } mint junretsu(int a,int b){ mint x = kaijou_[a]*kaijou_[b]; x *= kaijou[a+b]; return x; } mint catalan(int n){ return combination(2*n,n)/(n+1); } }; combi C(3000000); int main(){ int N,K; cin>>N>>K; vector<mint> dp(1,1); rep(i,K){ vector<mint> ndp(dp.size()+1,0); rep(j,dp.size()){ int two = j; int one = i - two*2; if(one<0)continue; int zero = N-one; if(zero<0)continue; ndp[j] += dp[j] * zero; ndp[j+1] += dp[j] * one; } swap(dp,ndp); } mint ans = 0; rep(i,dp.size()){ mint v = dp[i]; v *= N+i; ans += v; } ans /= mint(N).pow(K); cout<<ans.val()<<endl; return 0; }