結果
問題 | No.2484 Add to Variables |
ユーザー |
|
提出日時 | 2023-09-22 22:14:15 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 75 ms / 2,000 ms |
コード長 | 1,630 bytes |
コンパイル時間 | 1,896 ms |
コンパイル使用メモリ | 114,324 KB |
実行使用メモリ | 13,368 KB |
最終ジャッジ日時 | 2024-07-08 12:58:07 |
合計ジャッジ時間 | 4,016 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 39 |
ソースコード
#include<iostream>#include<cassert>#include<atcoder/modint>#include<atcoder/convolution>using namespace std;using mint=atcoder::modint998244353;#include<vector>template<typename T>struct combination{vector<T>fac,ifac;combination(size_t N=0):fac(1,1),ifac(1,1){make_table(N);}void make_table(size_t N){if(fac.size()>N)return;size_t now=fac.size();N=max(N,now*2);fac.resize(N+1);ifac.resize(N+1);for(size_t i=now;i<=N;i++)fac[i]=fac[i-1]*i;ifac[N]=1/fac[N];for(size_t i=N;i-->now;)ifac[i]=ifac[i+1]*(i+1);}T factorial(size_t n){make_table(n);return fac[n];}T invfac(size_t n){make_table(n);return ifac[n];}T P(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k];}T C(size_t n,size_t k){if(n<k)return 0;make_table(n);return fac[n]*ifac[n-k]*ifac[k];}T H(size_t n,size_t k){if(n==0)return k==0?1:0;return C(n-1+k,k);}};combination<mint>C;int N,M;int B[4];int main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin>>N>>M;for(int i=0;i<N;i++)cin>>B[i];vector<mint>dp(1,mint::raw(1));int base=B[0];int need=0;for(int i=0;i<3;i++){int d=B[i+1]-B[i];if(d<0)base-=-d;vector<mint>coef(M+1,mint::raw(0));d=abs(d);need+=d;for(int i=0;d+i*2<=M;i++){coef[d+i*2]=C.invfac(d+i)*C.invfac(i);}dp=atcoder::convolution(dp,coef);dp.resize(M+1);}mint ans=0;for(int i=need;i<=M;i+=2){int add=(i-need)/2;if(add<=base&&base-add<=M-i){ans+=dp[i]*C.invfac(base-add)*C.invfac(M-i-(base-add));}}cout<<(ans*C.factorial(M)).val()<<endl;}