Submission #1316034

#TimeUsernameProblemLanguageResultExecution timeMemory
1316034yus1f_mKpart (eJOI21_kpart)C++20
30 / 100
2095 ms480 KiB
#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#define ll long long
#define str string
#define pb push_back
#define pf push_front
#define in insert
#define all(v) v.begin(),v.end()
const int sz=50001,INF=1000000000;
using namespace std;
bitset<sz>dp;
bool f(ll num,vector<ll>nums)
{
    dp.reset();
    dp[0]=1;
    for(int i=0;i<nums.size();i++)
    {
        dp|=(dp<<nums[i]);
    }
    return dp[num/2];
}
void solve()
{
    ll n,num;
    bool flag;
    cin>>n;
    vector<ll>nums,prefs(n+1,0),Nums,ans;
    for(int i=0;i<n;i++)
    {
        cin>>num;
        nums.pb(num),prefs[i+1]=prefs[i]+num;
    }
    for(int i=2;i<=n;i++)
    {
        flag=false;
        for(int j=1;j<=n-i+1;j++)
        {
            if((prefs[j+i-1]-prefs[j-1])%2!=0)
            {
                flag=true;
                break;
            }
            else
            {
                Nums.clear();
                for(int k=j;k<=j+i-1;k++)
                {
                    Nums.pb(nums[k-1]);
                }
                if(!f(prefs[j+i-1]-prefs[j-1],Nums))
                {
                    flag=true;
                    break;
                }
            }
        }
        if(!flag)
        {
            ans.pb(i);
        }
    }
    cout<<ans.size()<<" ";
    for(int i=0;i<ans.size();i++)
    {
        cout<<ans[i]<<" ";
    }
    cout<<"\n";
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr),cout.tie(nullptr);
    ll t=1;
    cin>>t;
    while(t--)
    {
        solve();
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...