Submission #889573

#	Time	Username	Problem	Language	Result	Execution time	Memory
889573		8pete8	Kpart (eJOI21_kpart)	C++17	100 / 100	1873 ms	1496 KiB

Main

#include<iostream>
#include<stack>
#include<map>
#include<vector>
#include<string>
#include<unordered_map>
#include <queue>
#include<cstring>
#include<limits.h>
#include <cassert>
#include<cmath>
#include<set>
#include<algorithm>
#include <iomanip>
#include<numeric> //gcd(a,b)
#include<bitset>
using namespace std;
#define ll long long
#define f first
#define endl "\n"
#define s second
#define pii pair<int,int>
#define ppii pair<int,pii>
#define vi vector<int>
#define pb push_back
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()
#define F(n) for(int i=0;i<n;i++)
#define lb lower_bound
#define ub upper_bound
#define fastio ios::sync_with_stdio(false);cin.tie(NULL);
#pragma GCC optimize ("03,unroll-loops")
using namespace std;
#define int long long
const int mod=998244353,mxn=500+5,lg=30,inf=1e18,minf=-1e18,Mxn=100000;
void solve(){
    int n;cin>>n;
    vector<int>dp(Mxn+10,-1);
    vector<int>ans(n+1,1);
    vector<int>v(n+1);
    for(int i=1;i<=n;i++)cin>>v[i];
    int sum=0;
    for(int i=1;i<=n;i++){
        sum+=v[i];
        int sub=0;
        for(int j=Mxn;j>=0;j--){
            if(j-v[i]>=0&&dp[j-v[i]]!=-1)dp[j]=max(dp[j],dp[j-v[i]]);
            else if(j-v[i]==0)dp[j]=i;
        }
        sub=sum;
        for(int j=1;j<i;j++){
            if(sub%2==0&&dp[sub/2]>=j)ans[i-j+1]&=true;
            else ans[i-j+1]=false;
            sub-=v[j];
        }
    }
    int cnt=0;
    for(int i=2;i<=n;i++)if(ans[i])cnt++;
    cout<<cnt<<' ';
    for(int i=2;i<=n;i++)if(ans[i])cout<<i<<" ";
    cout<<'\n';
}
int32_t main(){
    fastio
    int t;cin>>t;
    while(t--)solve();
    return 0;
}

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