結果

問題 No.2164 Equal Balls
ユーザー umezoumezo
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
}
0