結果
問題 | No.2472 Tea time in the grand garden |
ユーザー |
![]() |
提出日時 | 2024-04-30 22:43:44 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 20 ms / 2,000 ms |
コード長 | 1,976 bytes |
コンパイル時間 | 2,063 ms |
コンパイル使用メモリ | 201,668 KB |
最終ジャッジ日時 | 2025-02-21 09:57:42 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
#include<bits/stdc++.h>#include<atcoder/modint>using namespace std;using mint=atcoder::modint998244353;template<typename mint>struct combination{combination(int n=0):inner_fac(1,1),inner_finv(1,1){init(n);}mint fac(int n){init(n);return inner_fac[n];}mint finv(int n){init(n);return inner_finv[n];}mint inv(int n){if(n==0)return 0;init(n);return inner_fac[n-1]*inner_finv[n];}mint C(int n, int r){if(r<0)return 0;if(n<0){n=-n;mint res=C(n-1+r,r);if(r&1)res=-res;return res;}if(n<r)return 0;if(n<bound){init(n);return inner_fac[n]*inner_finv[n-r]*inner_finv[r];}init(r);mint res=1;for(int i=0;i<r;i++)res*=(n-i);return res*inner_finv[r];}mint P(int n, int r){if(n<0||r<0||n<r)return 0;if(n<bound){init(n);return inner_fac[n]*inner_finv[n-r];}mint res=1;for(int i=0;i<r;i++)res*=(n-i);return res;}mint H(int n, int r){return C(n-1+r,r);}private:const int bound=1<<25;vector<mint>inner_fac,inner_finv;void init(int n){int sz=inner_fac.size();if(sz>n)return;n=min(max(n,2*sz),bound);inner_fac.resize(n+1);inner_finv.resize(n+1);for(int i=sz;i<=n;i++)inner_fac[i]=inner_fac[i-1]*i;inner_finv[n]=inner_fac[n].inv();for(int i=n;i>sz;i--)inner_finv[i-1]=inner_finv[i]*i;}};combination<mint>C;mint narayana(int n, int k){return C.C(n,k)*C.C(n,k-1)*C.inv(n);}int main(){int N,K;cin>>N>>K;mint ans=0;for(int j=1;j<=K;j++)ans+=C.C(2*K+N-2*j+1,N-2*j+1)*narayana(K,j);cout<<(K==0?mint(1):ans).val()<<endl;}