結果
問題 | No.2164 Equal Balls |
ユーザー |
![]() |
提出日時 | 2022-12-15 02:21:50 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,837 bytes |
コンパイル時間 | 2,305 ms |
コンパイル使用メモリ | 201,800 KB |
最終ジャッジ日時 | 2025-02-09 12:29:51 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 TLE * 33 |
ソースコード
#define rep(i,n) for(int i=0;i<(int)(n);i++) #define ALL(v) v.begin(),v.end() typedef long long ll; #include<bits/stdc++.h> using namespace std; const int MOD=998244353; const int MAX=610; ll fac[MAX],finv[MAX],inv[MAX]; void init(){ fac[0]=fac[1]=1; finv[0]=finv[1]=1; inv[1]=1; for(int i=2;i<MAX;i++){ fac[i]=fac[i-1]*i%MOD; inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD; finv[i]=finv[i-1]*inv[i]%MOD; } } ll nCr(int n,int k){ if(n<k) return 0; if(n<0 || k<0) return 0; return fac[n]*(finv[k]*finv[n-k]%MOD)%MOD; } int main(){ ios::sync_with_stdio(false); std::cin.tie(nullptr); init(); int n,m; cin>>n>>m; vector<ll> A(n),B(n); rep(i,n) cin>>A[i]; rep(i,n) cin>>B[i]; vector<vector<ll>> C(m,vector<ll>(601)); rep(i,m){ for(int j=0;j<=600;j++){ ll tmp=1; ll tj=j-300; for(int k=i;k<n;k+=m){ ll x; if(tj>=0){ if(tj>B[k]) x=0; else{ ll y=0; for(int l=0;l<=A[k] && l+tj<=B[k];l++){ y=(y+nCr(A[k],l)*nCr(B[k],l+tj)%MOD)%MOD; } x=y; } } else{ if(-tj>A[k]) x=0; else{ ll y=0; for(int l=0;l<=B[k] && l-tj<=A[k];l++){ y=(y+nCr(A[k],l-tj)*nCr(B[k],l)%MOD)%MOD; } x=y; } } tmp=tmp*x%MOD; } C[i][j]=tmp; } } vector<ll> dp(180180); int d=90000; for(int i=0;i<=600;i++){ int ti=i-300; dp[ti+d]=C[0][i]; } for(int i=0;i<m-1;i++){ vector<ll> dp1(180180); for(int j=45000;j<=135000;j++){ if(dp[j]==0) continue; for(int k=0;k<=600;k++){ int tk=k-300; dp1[j+tk]=(dp1[j+tk]+dp[j]*C[i+1][k]%MOD)%MOD; } } swap(dp,dp1); } cout<<dp[d]<<endl; return 0; }