Submission #1089198

#	Time	Username	Problem	Language	Result	Execution time	Memory
1089198		vjudge1	Kpart (eJOI21_kpart)	C++17	0 / 100	2075 ms	1116 KiB

Main

#include <bits/stdc++.h>
//qwerty47924692
using namespace std;
using ll = int;
const ll N=2e5+29;
ll n,t,a[N],mp[N];
void f1(ll x,ll n,ll s1=0,ll s2=0){
    if(x==n){
        if(s1<s2)swap(s1,s2);
        mp[s1-s2]=1;
        return;
    }
    x++;
    f1(x,n,s1+a[x],s2);
    f1(x,n,s1,s2+a[x]);
}
ll f2(ll x,ll n,ll s1=0,ll s2=0){
    if(x==n){
       // if(z==1)cout<<bool(mp[s1-s2])<<'\n';
       // if(mp[s1-s2])cout<<s1<<' '<<s2<<'\n';
       if(s1<s2)swap(s1,s2);
        return (mp[s1-s2]);
    }
    x++;
   // cout<<x<<' ';
    ll ans=max(f2(x,n,s1+a[x],s2),f2(x,n,s1,s2+a[x]));
    //if(z==1)cout<<ans;
    return ans;

}
void solve(){
    cin>>n;
    for(ll i=1;i<=n;i++){
        cin>>a[i];
    }
    vector<ll>v;
    for(ll k=1;k<=n;k++){
        for(ll i=0;i<N;i++)mp[i]=0;
        ll ch=1;

        for(ll i=1;i+k-1<=n;i++){
            ll mid=((i+k-1)+i)>>1;
            if(k==1&&a[i])ch=0;
            f1(i-1,mid);


            ch=min(ch,f2(mid,i+k-1));
           // if(k==1)cout<<f2(mid,i+k-1);

        }if(ch)v.push_back(k);

        ch=1;
    }
    cout<<v.size();
    for(ll i:v){
        cout<<' '<<i;
    }
}
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    ll t;

    cin>>t;
    while(t--){
        solve();
        cout<<'\n';
    }
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 20
• Subtask 3: 0 / 70