結果
| 問題 | No.2164 Equal Balls |
| ユーザー |
 ぷら
|
| 提出日時 | 2022-12-15 03:47:20 |
| 言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,859 bytes |
| コンパイル時間 | 4,985 ms |
| コンパイル使用メモリ | 268,780 KB |
| 実行使用メモリ | 27,564 KB |
| 最終ジャッジ日時 | 2024-04-26 07:12:47 |
| 合計ジャッジ時間 | 11,997 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge3 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 6 ms
8,388 KB |
| testcase_01 | AC | 6 ms
8,260 KB |
| testcase_02 | AC | 7 ms
8,516 KB |
| testcase_03 | AC | 6 ms
8,260 KB |
| testcase_04 | AC | 7 ms
8,264 KB |
| testcase_05 | TLE | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| 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 | -- | - |
ソースコード
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
constexpr int mod = 998244353;
long long fac[200005], finv[200005], inv[200005];
void COMinit() {
fac[0] = fac[1] = finv[0] = finv[1] = inv[1] = 1;
for (int i = 2; i < 200005; 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;
}
}
long long COM(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;
}
long long choose(int n,int k) {
if(n < 0 || k < 0) return 0;
if(n == 0) return 1;
return COM(n+k-1,k-1);
}
int main() {
COMinit();
ios::sync_with_stdio(false);
cin.tie(nullptr);
int N,M;
cin >> N >> M;
vector<int>A(N),B(N);
for(int i = 0; i < N; i++) {
cin >> A[i];
}
for(int i = 0; i < N; i++) {
cin >> B[i];
}
vector<vector<int>>dp(M,vector<int>(601,1));
for(int i = 0; i < N; i++) {
vector<int>a(A[i]+1),b(B[i]+1);
for(int j = 0; j <= A[i]; j++) {
a[j] = COM(A[i],j);
}
for(int j = 0; j <= B[i]; j++) {
b[j] = COM(B[i],j);
}
reverse(b.begin(),b.end());
vector<int>c = convolution(a,b);
vector<int>d(601);
for(int j = 0; j <= A[i]+B[i]; j++) {
d[300+j-B[i]] = c[j];
}
for(int j = 0; j <= 600; j++) {
dp[i%M][j] = 1ll*dp[i%M][j]*d[j]%mod;
}
}
while(dp.size() > 1) {
vector<vector<int>>nxt;
for(int i = 0; i < dp.size(); i += 2) {
nxt.push_back(convolution(dp[i],dp[i+1]));
}
if(dp.size()%2) {
nxt.push_back(dp.back());
}
dp = nxt;
}
cout << dp[0][300*M] << endl;
}