Submission #379898

#	Time	Username	Problem	Language	Result	Execution time	Memory
379898		Kalashnikov	Bootfall (IZhO17_bootfall)	C++17	65 / 100	685 ms	5996 KiB

bootfall

#include <bits/stdc++.h>
 
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define all(a) a.begin() , a.end()
#define F first
#define S second
 
using namespace std;
using ll = long long;
 
const int N = 500+5 , inf = 2e9 + 7;
const ll INF = 1e18 ,   mod = 1e9+7 , P = 6547;
 
int a[N] , u[N];
bitset<500*250> dp[N] , b1;
 
 
void solve() {
	int n;
	cin >> n;
	int sum = 0;
	int cnt[2];
	cnt[0]=cnt[1]=0;
	for(int i = 1; i <= n; i ++) {
		cin >> a[i];
		sum += a[i];
		cnt[a[i]%2] = 1;
	}
	set<int> st;
	for(int i = 0; i <= n; i ++) {
		if(u[a[i]]) continue;
		u[a[i]] = 1;
		for(int j = 1; j <= n; j ++) {
			if(i == j) continue;
			dp[i] = (dp[i] | (dp[i] << a[j]));
			dp[i][a[j]] = 1;
		}
		dp[i] <<= a[i]/2;
		if(i == 1) b1 = dp[i];
		if(i > 1) b1 &= dp[i];
	}
	if(sum%2 || !dp[0][sum/2] || cnt[0] == cnt[1]) {
		cout << 0;
		return;
	}
	vector<int> ans;
	int sta = 0;
	if(cnt[1]) sta = 1;
	for(int i = sta; i <= sum; i += 2) {
		int cur = (sum+i)/2;
		if(b1[cur]) ans.push_back(i);
	}
	sort(all(ans));
	cout << ans.size() << '\n';
	for(auto to: ans) {
		cout << to << ' ';
	}
}
/*
1 2 1 2   2
 
 
*/
main() {
    ios;
    int tt = 1;
    //cin >> tt;
    while(tt --) {
        solve();
    }
    return 0;
}

Compilation message (stderr)

bootfall.cpp:65:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   65 | main() {
      |      ^

• Subtask 1: 6 / 6
• Subtask 2: 7 / 7
• Subtask 3: 15 / 15
• Subtask 4: 16 / 16
• Subtask 5: 21 / 21
• Subtask 6: 0 / 35