Submission #1056051

#	Time	Username	Problem	Language	Result	Execution time	Memory
1056051		alex_2008	Kpart (eJOI21_kpart)	C++14	30 / 100	2037 ms	484 KiB

Main

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#include <bitset>
typedef long long ll;
using namespace std;
const int N = 1e5 + 10;
int a[N], cnt[N];
bitset <N> Used;
void solve() {
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		cnt[i - 1] = 0;
	}
	for (int i = 1; i <= n; i++) {
		Used.reset();
		Used[0] = 1;
		int cur = 0;
		for (int j = i; j <= n; j++) {
			Used |= (Used << a[j]);
			cur += a[j];
			if (cur % 2 == 0 && Used[cur / 2]) cnt[j - i]++;
		}
	}
	vector <int> ans;
	for (int i = 0; i < n; i++) {
		if (cnt[i] == n - i) ans.push_back(i + 1);
	}
	cout << ans.size() << " ";
	for (auto it : ans)
		cout << it << " ";
	cout << "\n";
}
int main() {
	int t;
	cin >> t;
	while (t--)
		solve();
}

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