Submission #537521

#	Time	Username	Problem	Language	Result	Execution time	Memory
537521		zaneyu	Kpart (eJOI21_kpart)	C++14	100 / 100	1578 ms	1008 KiB

Main

/*input
2
7
7 3 5 1 3 3 5
6
1 2 3 5 8 3
*/
#include<bits/stdc++.h>
using namespace std;
#define REP(i,n) for(int i=0;i<n;i++)
#define MNTO(x,y) x=min(x,y)
#define MXTO(x,y) x=max(x,y)
#define REP1(i,n) for(int i=1;i<=n;i++)
#define ll long long
#define ld long double
#define sz(x) (int)x.size()
#define pb push_back
const int maxn=1e3+5;
int pref[maxn],arr[maxn];
int dp[50005],ndp[50005];
bool wk[maxn];
void solve(){
    int n;
    cin>>n;
    REP1(i,n) cin>>arr[i],pref[i]=pref[i-1]+arr[i],wk[i]=1;
    REP(j,50005) dp[j]=0;
    dp[0]=1;
    REP1(i,n){
        ndp[0]=i+1;
        REP1(j,50004){
            ndp[j]=dp[j];
            if(j>=arr[i]) MXTO(ndp[j],dp[j-arr[i]]);
        }
        REP1(j,i){
            int x=pref[i]-pref[j-1];
            if(x%2 or ndp[x/2]<j) wk[i-j+1]=0;
        }
        REP(j,50005) dp[j]=ndp[j];
    }
    int cnt=0;
    REP1(i,n) cnt+=wk[i];
    cout<<cnt<<' ';
    REP1(i,n) if(wk[i]) cout<<i<<' ';
    cout<<'\n';
}
int main(){
    ios::sync_with_stdio(false),cin.tie(0);
    int t;
    cin>>t;
    while(t--) solve();
}

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