結果
問題 | No.2062 Sum of Subset mod 999630629 |
ユーザー | ぷら |
提出日時 | 2022-08-26 22:36:40 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 1,850 ms / 5,000 ms |
コード長 | 1,664 bytes |
コンパイル時間 | 2,002 ms |
コンパイル使用メモリ | 202,020 KB |
実行使用メモリ | 11,400 KB |
最終ジャッジ日時 | 2023-08-04 04:01:03 |
合計ジャッジ時間 | 7,158 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 6 ms
9,180 KB |
testcase_01 | AC | 5 ms
9,176 KB |
testcase_02 | AC | 6 ms
9,332 KB |
testcase_03 | AC | 6 ms
9,228 KB |
testcase_04 | AC | 6 ms
9,228 KB |
testcase_05 | AC | 5 ms
9,392 KB |
testcase_06 | AC | 6 ms
9,328 KB |
testcase_07 | AC | 6 ms
9,368 KB |
testcase_08 | AC | 28 ms
9,244 KB |
testcase_09 | AC | 24 ms
9,308 KB |
testcase_10 | AC | 22 ms
9,308 KB |
testcase_11 | AC | 53 ms
10,092 KB |
testcase_12 | AC | 64 ms
10,280 KB |
testcase_13 | AC | 39 ms
9,804 KB |
testcase_14 | AC | 63 ms
10,236 KB |
testcase_15 | AC | 31 ms
9,480 KB |
testcase_16 | AC | 47 ms
9,876 KB |
testcase_17 | AC | 54 ms
10,020 KB |
testcase_18 | AC | 39 ms
9,816 KB |
testcase_19 | AC | 31 ms
9,356 KB |
testcase_20 | AC | 35 ms
9,692 KB |
testcase_21 | AC | 35 ms
9,540 KB |
testcase_22 | AC | 34 ms
9,660 KB |
testcase_23 | AC | 19 ms
9,376 KB |
testcase_24 | AC | 19 ms
9,500 KB |
testcase_25 | AC | 194 ms
11,312 KB |
testcase_26 | AC | 294 ms
11,400 KB |
testcase_27 | AC | 241 ms
11,344 KB |
testcase_28 | AC | 235 ms
11,284 KB |
testcase_29 | AC | 136 ms
11,084 KB |
testcase_30 | AC | 109 ms
10,840 KB |
testcase_31 | AC | 1,850 ms
10,720 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; constexpr int mod1 = 998244353; constexpr int mod2 = 999630629; long long modpow(long long a,long long b,long long m) { long long ans = 1; while(b) { if(b & 1) { (ans *= a) %= m; } (a *= a) %= m; b /= 2; } return ans; } 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 % mod1; inv[i] = mod1 - inv[mod1 % i] * (mod1 / i) % mod1; finv[i] = finv[i - 1] * inv[i] % mod1; } } 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] % mod1) % mod1; } int dp[2][404000]; inline void mpl(int &x,int y) { x += y; if(x >= mod1) x -= mod1; } int main() { COMinit(); int N; cin >> N; vector<int>A(N),B(10010); int sum = 0; for(int i = 0; i < N; i++) { cin >> A[i]; sum += A[i]; B[A[i]]++; } int ans = sum*modpow(2,N-1,mod1)%mod1; dp[0][0] = 1; for(int i = 1; i <= 10000; i++) { if(!B[i]) continue; for(int j = 0; j <= sum-mod2; j++) { for(int k = 0; j+k*i <= sum-mod2 && k <= B[i]; k++) { mpl(dp[1][j+k*i],1ll*dp[0][j]*COM(B[i],k)%mod1); } } for(int j = 0; j <= sum-mod2; j++) { dp[0][j] = dp[1][j]; dp[1][j] = 0; } } for(int i = 0; i <= sum-mod2; i++) { mpl(ans,mod1-1ll*(mod2-mod1)*dp[0][i]%mod1); } cout << ans << endl; }