結果

問題 No.2164 Equal Balls
ユーザー umezoumezo
提出日時 2022-12-15 02:21:50
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,837 bytes
コンパイル時間 2,410 ms
コンパイル使用メモリ 204,228 KB
実行使用メモリ 14,060 KB
最終ジャッジ日時 2024-04-26 06:03:42
合計ジャッジ時間 14,146 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
5,888 KB
testcase_01 AC 4 ms
6,144 KB
testcase_02 AC 7 ms
6,140 KB
testcase_03 AC 4 ms
6,016 KB
testcase_04 AC 6 ms
6,148 KB
testcase_05 AC 6 ms
6,280 KB
testcase_06 AC 7 ms
6,280 KB
testcase_07 AC 6 ms
6,152 KB
testcase_08 AC 2,080 ms
6,804 KB
testcase_09 AC 2,151 ms
6,696 KB
testcase_10 AC 425 ms
6,392 KB
testcase_11 TLE -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
testcase_45 -- -
testcase_46 -- -
testcase_47 -- -
testcase_48 -- -
testcase_49 -- -
testcase_50 -- -
testcase_51 -- -
testcase_52 -- -
testcase_53 -- -
権限があれば一括ダウンロードができます

ソースコード

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