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>
#define boost ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
const int N=2e5+5;
const int inf=1e9;
const int mod=1e9+7;
const int N1=1e5+25;
vector<int>v;
vector<int>v1;
struct edge{
int v,time;
};
int dp[N1];
int n,m;
set<int>st;
signed main(){
// freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
boost;
int t;
cin>>t;
while(t--){
int n;
cin>>n;
v.push_back(0);
v1.push_back(0);
int cnt1=0;
for(int i=0;i<n;i++){
int x;
cin>>x;
cnt1+=x;
v.push_back(x);
v1.push_back(cnt1);
if(i>=1){
st.insert(i+1);
}
}
int cnt=0;
for(int i=1;i<=n;i++){
for(int j=N1-1;j>v[i];j--){
dp[j]=max(dp[j],dp[j-v[i]]);
}
dp[v[i]]=i;
for(int j=2;j<=i;j++){
int cnt=v1[i]-v1[i-j];
if(dp[cnt/2]<=i-j||cnt%2==1){
st.erase(j);
}
}
}
cout<<st.size()<<" ";
for(auto it:st){
cout<<it<<" ";
}
for(int i=0;i<N1;i++){
dp[i]=0;
}
cout<<"\n";
v.clear();
v1.clear();
}
}
Compilation message (stderr)
Main.cpp: In function 'int main()':
Main.cpp:39:13: warning: unused variable 'cnt' [-Wunused-variable]
39 | int cnt=0;
| ^~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |