#include <bits/stdc++.h> #define PB push_back #define ST first #define ND second #define _ ios_base::sync_with_stdio(0); cin.tie(0); //mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count()); using namespace std; using ll = long long; using pi = pair<int,int>; using vi = vector<int>; const int nax = 5000 + 10, mod = 1e9 + 7; int n; int a[nax]; int dp[nax][nax]; int mx; int main() {_ cin >> n; for(int i = 1; i <= n; ++i) { cin >> a[i]; mx = max(mx, a[i]); } sort(a + 1, a + n + 1); int ans = 0; dp[0][0] = 1; for(int i = 1; i <= n; ++i) { ans = (ans + ans) % mod; for(int s = 0; s <= mx; ++s) { dp[i][s] = dp[i - 1][s]; if(s >= a[i] && s - a[i] >= a[i] - 1) { dp[i][s] = (dp[i][s] + dp[i - 1][s - a[i]]) % mod; } if(s + a[i] > mx && s >= a[i] - 1) { ans = (ans + dp[i - 1][s]) % mod; } } } for(int s = 0; s <= mx; ++s) { ans = (ans + dp[n][s]) % mod; } ans--; if(ans < 0) ans += mod; cout << ans; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | #include <bits/stdc++.h> #define PB push_back #define ST first #define ND second #define _ ios_base::sync_with_stdio(0); cin.tie(0); //mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count()); using namespace std; using ll = long long; using pi = pair<int,int>; using vi = vector<int>; const int nax = 5000 + 10, mod = 1e9 + 7; int n; int a[nax]; int dp[nax][nax]; int mx; int main() {_ cin >> n; for(int i = 1; i <= n; ++i) { cin >> a[i]; mx = max(mx, a[i]); } sort(a + 1, a + n + 1); int ans = 0; dp[0][0] = 1; for(int i = 1; i <= n; ++i) { ans = (ans + ans) % mod; for(int s = 0; s <= mx; ++s) { dp[i][s] = dp[i - 1][s]; if(s >= a[i] && s - a[i] >= a[i] - 1) { dp[i][s] = (dp[i][s] + dp[i - 1][s - a[i]]) % mod; } if(s + a[i] > mx && s >= a[i] - 1) { ans = (ans + dp[i - 1][s]) % mod; } } } for(int s = 0; s <= mx; ++s) { ans = (ans + dp[n][s]) % mod; } ans--; if(ans < 0) ans += mod; cout << ans; } |