Submission #891614

#	Time	Username	Problem	Language	Result	Execution time	Memory
891614		imarn	Kpart (eJOI21_kpart)	C++14	0 / 100	275 ms	528 KiB

Main

#include "bits/stdc++.h"
#define f first
#define s second
#define ll long long
#define pb push_back
#define pii pair<ll,ll>
#define all(x) x.begin(),x.end()
#define sz(x) (int)x.size()
#define vi vector<int>
#define vvi vector<vi>
using namespace std;
void solve(){
    int n;cin>>n;
    int a[n+1];for(int i=1;i<=n;i++)cin>>a[i];
    int qs[n+1]={0};for(int i=1;i<=n;i++)qs[i]=qs[i-1]+a[i];
    bool vis[100001]={0};
    bool ch[n+1]={0};ch[1]=1;
    int nw=0,pr=0;vis[0]=1;int cnt=n-1;
    for(int i=1;i<=n;i++){
        pr=nw;nw=1-nw;
        for(int j=100000;j>=0;j--)if(vis[j]&&j+a[i]<=100000)vis[j+a[i]]=1;
        for(int j=2;j<=i;j++){
            int x=qs[i]-qs[i-j];
            if(x&1&&!ch[j]){ch[j]=1,cnt--;continue;}x>>=1;
            if(!(vis[x]||vis[x+qs[i-j]])&&!ch[j])ch[j]=1,cnt--;
        }
    }cout<<cnt<<" ";
    for(int i=1;i<=n;i++)if(!ch[i])cout<<i<<" ";cout<<'\n';
}
int main(){
    ios_base::sync_with_stdio(0);cin.tie(0);
    int t;cin>>t;while(t--)solve();
}

Compilation message (stderr)

Main.cpp: In function 'void solve()':
Main.cpp:28:5: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
   28 |     for(int i=1;i<=n;i++)if(!ch[i])cout<<i<<" ";cout<<'\n';
      |     ^~~
Main.cpp:28:49: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
   28 |     for(int i=1;i<=n;i++)if(!ch[i])cout<<i<<" ";cout<<'\n';
      |                                                 ^~~~
Main.cpp:18:14: warning: variable 'pr' set but not used [-Wunused-but-set-variable]
   18 |     int nw=0,pr=0;vis[0]=1;int cnt=n-1;
      |              ^~

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