제출 #91470

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
91470		davitmarg	Bootfall (IZhO17_bootfall)	C++17	13 / 100	3 ms	740 KiB

bootfall

/*
DEATH-MATCH
Davit-Marg
*/
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <stack>
#include <iterator>
#include <ctype.h>
#include <stdlib.h>  
#include <fstream>  
#define mod 1000000007ll
#define LL long long
#define LD long double
#define MP make_pair
#define PB push_back
using namespace std;

LL n,a[5000006],cant[5000006];
LL sum,sm;
vector<LL> ans;
LL dp[5000006];

void add(LL k)
{
	for (LL i = sum; i >= 0; i--) {
		dp[i+k] += dp[i];
	}
}

void rem(LL k)
{
	for (LL i = 0; i <= sum; i++) {
		dp[i+k] -= dp[i];
	}
}

int main()
{
	cin >> n;
	for (LL i = 0; i < n; i++)
	{
		cin >> a[i];
		sum += a[i];
	}
	dp[0] = 1ll;

	for (LL i = 0; i < n; i++)
		for (LL j = sum; j >= 0; j--)
			dp[j + a[i]]+=dp[j];

	if (sum % 2==1 || dp[sum / 2] == 0)
	{
		cout << 0 << endl;
		return 0;
	}
	sm = sum;
	for (LL i = 0; i < n; i++)
	{
		sum -= a[i];
		rem(a[i]);
		for (LL j = 0; j <= sm*2; j++)
			if ((sum+j) % 2==1 || dp[(sum+j) / 2] <= 0)
				cant[j] = 1;
		add(a[i]);
		sum += a[i];
	}
	for (LL j = 1; j <= sm; j++)
		if (!cant[j])
			ans.PB(j);
	
	cout << ans.size() << endl;
	if (ans.size() == 0)
		return 0;
	for (LL i = 0; i < ans.size(); i++)
		cout << ans[i] << " ";
	cout << endl;
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

bootfall.cpp: In function 'int main()':
bootfall.cpp:83:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (LL i = 0; i < ans.size(); i++)
                 ~~^~~~~~~~~~~~

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