Submission #1316029

#TimeUsernameProblemLanguageResultExecution timeMemory
1316029yus1f_mKpart (eJOI21_kpart)C++20
Compilation error
0 ms0 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 ll sz=6000001,INF=1000000000;
using namespace std;
bitset<sz>dp;
bool f(vector<ll>nums)
{
    dp.reset();
    dp[0]=1;
    for(int i=0;i<nums.size();i++)
    {
        dp|=(dp<<nums[i]);
    }
    return dp[sum/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(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();
    }
}

Compilation message (stderr)

Main.cpp: In function 'bool f(std::vector<long long int>)':
Main.cpp:20:15: error: 'sum' was not declared in this scope
   20 |     return dp[sum/2];
      |               ^~~