Submission #888397

#	Time	Username	Problem	Language	Result	Execution time	Memory
888397		pcc	Bootfall (IZhO17_bootfall)	C++14	65 / 100	747 ms	4616 KiB

bootfall

#include <bits/stdc++.h>
using namespace std;
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,popcnt")

#define ll long long
#define pll pair<ll,ll>
#define pii pair<int,int>
#define fs first
#define sc second
#define tlll tuple<ll,ll,ll>
mt19937 seed(time(NULL));
#define rnd(l,r) uniform_int_distribution<int>(l,r)(seed)

const int mxn = 510;
const int C = mxn*mxn/2;
int mod = 998244353;
int arr[mxn];
int dp[C];
int del[C];
int n;
int sum = 0;
bitset<C> able;
set<int> ans;

inline int mad(int a,int b){
	a += b;
	return a>=mod?a-mod:a;
}
inline int mub(int a,int b){
	return mad(a,mod-b);
}

inline bool check(int now,int tsum){
	if(sum&1)return false;
	if(!dp[sum>>1])return false;
	if(tsum&1)return false;
	if(now+now>tsum)return false;
	if(!del[tsum>>1])return false;
	return true;
}

inline void relax(int tar){
	mod = tar;
	memset(dp,0,sizeof(dp));
	dp[0] = 1;
	able.set();
	for(int i = 1;i<=n;i++){
		for(int j = C-1;j>=arr[i];j--)dp[j] = mad(dp[j],dp[j-arr[i]]);
	}
	for(int i = 1;i<=n;i++){
		for(int j = 0;j<C;j++)del[j] = dp[j];
		for(int j = arr[i];j<C;j++)del[j] = mub(del[j],del[j-arr[i]]);
		for(int j = 1;j<=sum;j++){
			if(!check(j,sum-arr[i]+j))able[j] = false;
		}
	}
	for(int j = 1;j<=sum;j++)if(able[j])ans.insert(j);
}

int main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n;
	dp[0] = 1;
	for(int i = 1;i<=n;i++){
		cin>>arr[i];
		sum += arr[i];
	}
	int lim = CLOCKS_PER_SEC*0.9;
	int st = clock();

	relax((int)1e9+7);
	relax((int)712271227);
	relax((int)1e8+7);
	//relax(113);
	relax(998244353);
	relax(101010111);
	const int inf = 1e9+10;
	//cout<<rnd(0,inf);
	//if(n>350)while(clock()-st<=lim)relax(rnd(0,inf));


	/*
	for(int i = 0;i<=10;i++)cout<<dp[i]<<' ';cout<<endl;
	cout<<sum<<endl;
	cout<<endl;
	for(int i = 1;i<=n;i++){
		for(int j = 0;j<=10;j++)cout<<del[i][j]<<' ';cout<<endl;
	}
	sort(ans.begin(),ans.end());
	ans.erase(unique(ans.begin(),ans.end()),ans.end());
   */

	cout<<ans.size()<<'\n';
	for(auto &i:ans)cout<<i<<' ';
	return 0;
}

Compilation message (stderr)

bootfall.cpp: In function 'int main()':
bootfall.cpp:69:6: warning: unused variable 'lim' [-Wunused-variable]
   69 |  int lim = CLOCKS_PER_SEC*0.9;
      |      ^~~
bootfall.cpp:70:6: warning: unused variable 'st' [-Wunused-variable]
   70 |  int st = clock();
      |      ^~
bootfall.cpp:78:12: warning: unused variable 'inf' [-Wunused-variable]
   78 |  const int inf = 1e9+10;
      |            ^~~

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