This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "bits/stdc++.h"
using namespace std;
#define ff first
#define ss second
#define all(v) v.begin(), v.end()
#define ll long long
#define pb push_back
#define pii pair<int, int>
#define pli pair<ll, int>
#define pll pair<ll, ll>
#define tr(i, c) for(auto i = c.begin(); i != c.end(); ++i)
#define wr puts("----------------")
template<class T>bool umin(T& a,T b){if(a>b){a=b;return 1;}return 0;}
template<class T>bool umax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
const int N = 1003;
const int S = 1e5+5;
int v[N], par[N], ndp[S], dp[S];
bool ok[N];
void solve(int green_flecks){
int n;
scanf("%d", &n);
for(int i = 1; i <= n; ++i){
int x;
scanf("%d", &x);
v[i] = x, ok[i] = 1, par[i] = par[i-1]+x, dp[i] = 0, ndp[i] = 0;
}
int ss = accumulate(v+1, v+n+1, 0);
for(int i = 1; i <= n; i++){
for(int j = 0; j <= ss; ++j)
ndp[j] = max(dp[j], dp[j-v[i]]);
ndp[v[i]] = i;
for(int j = 0; j <= ss; ++j)
dp[j] = ndp[j];
for(int j = i-1; j >= 1; j--){
int sm = par[i]-par[j-1], k = i-j+1;
if(sm&1 or dp[sm/2]<j)
ok[k] = 0;
}
}
vector<int> A;
for(int i = 2; i <= n; ++i)
if(ok[i])
A.pb(i);
printf("%d", (int)A.size());
tr(it, A)
printf(" %d", *it);
puts("");
}
int main(){
int tt;
scanf("%d", &tt);
while(tt--)
solve(tt);
return 0;
}
Compilation message (stderr)
Main.cpp: In function 'void solve(int)':
Main.cpp:22:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
22 | scanf("%d", &n);
| ~~~~~^~~~~~~~~~
Main.cpp:25:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
25 | scanf("%d", &x);
| ~~~~~^~~~~~~~~~
Main.cpp: In function 'int main()':
Main.cpp:53:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
53 | scanf("%d", &tt);
| ~~~~~^~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |