//Bismillahir-Rahmanir-Rahim
#include <bits/stdc++.h>
#pragma comment(linker, "/stack:200000000")
#pragma GCC optimize("O1,O2,O3,Ofast,unroll-loops")
#pragma GCC target("sse,sse2,sse3,sse4,sse4.1,sse4.2,popcnt,avx,avx2")
#define pb push_back
#define pii pair <int, int>
#define pll pair <long long, long long>
#define pld pair <long double, long double>
#define ll long long
#define ld long double
#define x first
#define y second
#define all(v) v.begin(),v.end()
#define sz(s) (int)s.size()
#define skip continue
#define bpop(x) (ll)__builtin_popcountll(x)
using namespace std;
const int N = 5e2 + 7;
const int W = 3e5 + 7;
const int maxA = 1e5 + 7;
const int inf = 1e9 + 7;
const ll INF = 2e18 + 7;
const int MOD = 998244353;
const ld eps = 1e-9;
pii dir[] = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
//mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
//clock_t start = clock();
//#define int long long
int n, a[N], dp[W];
void add(int x) {
for (int w = W - 1;w >= 0;w--) {
if (w + x < W)dp[w + x] += dp[w];
}
dp[x]++;
}
void del(int x) {
dp[x]--;
for (int w = x;w < W;w++)dp[w] -= dp[w - x];
}
bool check(int Tima) {
for (int w = 0;w < W;w++)dp[w] = 0;
int sum = 0;
for (int i = 1;i <= n;i++)add(a[i]), sum += a[i];
if (sum % 2 == 1 || !dp[sum / 2])return 0;
add(Tima), sum += Tima;
for (int i = 1;i <= n;i++) {
sum -= a[i], del(a[i]);
if (sum % 2 == 1 || !dp[sum / 2])return 0;
sum += a[i], add(a[i]);
}
return 1;
}
int cnt[W];
void solve() {
cin >> n;
for (int i = 1;i <= n;i++)cin >> a[i];
for (int i = 1;i <= n;i++) {
if (a[i] % 2 != a[1] % 2)cout << 0, exit(0);
}
for (int w = 0;w < W;w++)dp[w] = 0;
int sum = 0;
for (int i = 1;i <= n;i++)add(a[i]), sum += a[i];
for (int i = 1;i <= n;i++) {
sum -= a[i], del(a[i]);
for (int w = 1;w < W;w++) {
if ((w + sum) % 2 != 0)skip;
if ((sum + w) / 2 - w < 0)skip;
if (dp[(sum + w) / 2 - w] || (sum + w) / 2 - w == 0)cnt[w]++;
}
sum += a[i], add(a[i]);
}
vector <int> ans;
for (int w = 0;w < W;w++) {
if (cnt[w] == n)ans.pb(w);
}
cout << sz(ans) << '\n';
for (auto w : ans)cout << w << ' ';
}
signed main() {
//srand(time(NULL));
ios_base::sync_with_stdio(0);
cin.tie(0);
//freopen("subsequence.in", "r", stdin);
//freopen("subsequence.out", "w", stdout);
int test = 1;
//cin >> test;
for (int t = 1;t <= test;t++) {
//cout << "Case " << t << ": ";
solve();
}
return 0;
}
Compilation message
bootfall.cpp:4: warning: ignoring '#pragma comment ' [-Wunknown-pragmas]
4 | #pragma comment(linker, "/stack:200000000")
|
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
2396 KB |
Output is correct |
2 |
Correct |
4 ms |
2396 KB |
Output is correct |
3 |
Incorrect |
3 ms |
2396 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
2396 KB |
Output is correct |
2 |
Correct |
4 ms |
2396 KB |
Output is correct |
3 |
Incorrect |
3 ms |
2396 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
2396 KB |
Output is correct |
2 |
Correct |
4 ms |
2396 KB |
Output is correct |
3 |
Incorrect |
3 ms |
2396 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
2396 KB |
Output is correct |
2 |
Correct |
4 ms |
2396 KB |
Output is correct |
3 |
Incorrect |
3 ms |
2396 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
2396 KB |
Output is correct |
2 |
Correct |
4 ms |
2396 KB |
Output is correct |
3 |
Incorrect |
3 ms |
2396 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
2396 KB |
Output is correct |
2 |
Correct |
4 ms |
2396 KB |
Output is correct |
3 |
Incorrect |
3 ms |
2396 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |