Submission #1112911

#	Time	Username	Problem	Language	Result	Execution time	Memory
1112911		adiyer	Bootfall (IZhO17_bootfall)	C++17	65 / 100	1051 ms	1912 KiB

bootfall

#include <bits/stdc++.h>
 
#define ios ios_base::sync_with_stdio(0); cin.tie(0);
#define pb push_back
 
using namespace std;
 
typedef int ll;
 
const int N = 5e2 + 11;
const int MAX = 1e6;
 
short int n;

int s;

short int a[N];

vector < int > ans;
 
bitset < 500 * 500 + 3 > dp;

void sol(){
	cin >> n, dp[0] = 1;
	for(ll i = 1; i <= 500 * 500; i++) ans.pb(i);
	for(ll i = 1; i <= n; i++) cin >> a[i], dp |= (dp << a[i]), s += a[i];
	if(s % 2 || dp[s / 2] == 0){
		cout << "0\n";
		return;
	}
	for(short int i = 1; i <= n; i++){
		dp &= 0, dp[0] = 1, s = 0;
		for(short int j = 1; j <= n; j++)
			if(i != j)
				dp |= (dp << a[j]), s += a[j];
		int j = 0;
		while(j < ans.size()){
			if(ans.size() && (s < ans[j] || (s - ans[j]) % 2 || !dp[(s - ans[j]) / 2])){
				if(j < ans.size()) swap(ans[j], ans[ans.size() - 1]);
				ans.pop_back();
			}
			else{
				j++;
			}
		}
	}
	sort(ans.begin(), ans.end());
	cout << ans.size() << '\n';
	for(int x : ans) cout << x << ' ';
}
 
signed main(){
	ios                     
	sol();
// 	slow();
//  stress();
}

Compilation message (stderr)

bootfall.cpp: In function 'void sol()':
bootfall.cpp:37:11: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |   while(j < ans.size()){
      |         ~~^~~~~~~~~~~~
bootfall.cpp:39:10: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |     if(j < ans.size()) swap(ans[j], ans[ans.size() - 1]);
      |        ~~^~~~~~~~~~~~

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