Submission #1349115

#TimeUsernameProblemLanguageResultExecution timeMemory
1349115Faisal_SaqibKpart (eJOI21_kpart)C++20
100 / 100
832 ms1268 KiB
#include <iostream>
using namespace std;
const int N=1e5+10;
int a[N],pos[N];
int dp[N];
void solve()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	for(int i=0;i<N;i++)
	{
		pos[i]=1;
		dp[i]=1e9;
	}
	dp[0]=0;
	for(int i=n;i>=1;i--)
	{
		for(int j=N-1;j>=a[i];j--)
		{
			dp[j]=min(dp[j],max(i,dp[j-a[i]]));
		}
		int sm=0;
		for(int j=i;j<=n;j++)
		{
			sm+=a[j];
			if(sm%2==0 and dp[sm/2]<=j)
			{

			}
			else
			{
				pos[j-i]=0;
			}
		}
	}
	int ans=0;
	for(int i=0;i<n;i++)ans+=pos[i];
	cout<<ans<<' ';
	for(int i=0;i<n;i++)
	{
		if(pos[i])
		{
			cout<<i+1<<' ';
		}
	}
	cout<<endl;
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int t;
	cin>>t;
	while(t--)
	{
		solve();
	}
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...