Submission #840921

#	Time	Username	Problem	Language	Result	Execution time	Memory
840921		pera	Kpart (eJOI21_kpart)	C++17	30 / 100	2072 ms	193924 KiB

Main

#include<bits/stdc++.h>
using namespace std;

//#define int long long

const int N = 2e5 + 1;

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	int t = 1;cin >> t;
	while(t --){
		int n;cin >> n;
		vector<int> a(n + 1) , p(n + 1) , ls(n + 1) , ans;
		for(int i = 1;i <= n;i ++){
			cin >> a[i];
			p[i] = p[i - 1] + a[i];
			ls[i] = 1;
		}
		int X = p[n] / 2;
		vector<vector<int>> dp(n + 1 , vector<int>(X));
		for(int i = 1;i <= n;i ++){
			dp[i] = dp[i - 1];
			dp[i][a[i]] = i;
			for(int j = X;j > a[i];j --){
				dp[i][j] = max(dp[i][j] , dp[i - 1][j - a[i]]);
			}
		}
		for(int i = 1;i <= n;i ++){
			for(int j = i;j <= n;j ++){
				ls[j - i + 1] &= ((p[j] - p[i - 1]) % 2 == 0 && dp[j][(p[j] - p[i - 1]) / 2] >= i);
			}
		}
		cout << accumulate(ls.begin() , ls.end() , 0) << endl;
		for(int i = 1;i <= n;i ++){
			if(ls[i] == 1) cout << i << " ";
		}
		cout << endl;
	}
}

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