C. Baby Ehab Partitions Again
题目大意
给你一个长度为n的序列要求你使得这个序列无论怎么划分成两个子序列,其和都不会相等。要求你输出这个序列中需要删除元素的下标。
思路
当原序列的和为奇数时不需要删除,这个序列无论怎么划分都不会出现相等的情况,如果为偶数,先除去序列元素的最大公因子,这样方便找出一个奇数,直接去除,然后就是判断这些元素中能不能凑出总和的一半,能不能凑出一半,等价于01背包问题,背包的限制就是总和的一半,最大价值也就是总和的一半。
通过代码
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")
using namespace std;
#define ll long long
#define sl(n) scanf("%lld",&n)
#define pl(n) printf("%lld",n)
#define sdf(n) scanf("%lf",&n)
#define pdf(n) printf("%.lf",n)
#define pE printf("\n")
#define ull unsigned long long
#define pb push_back
#define debug(a) cout<<a<<"??"
#define me(a) memset(a,0,sizeof(a))
#define pre(beg,en) for(ll i=beg;i<=en;i++)
#define rep(beg,en) for(ll i=beg;i>=en;i--)
#define ph push
#define pi pair<ll,ll>
#define fi first
#define se second
const ll mod = 1e9 + 7;
ll a[3000];
ll dp[200010];
int main(){
ll n,sum=0,sum1=0;
sl(n);
pre(1,n)sl(a[i]);
ll temp=a[1];
pre(2,n)temp=__gcd(temp,a[i]);
pre(1,n){
a[i]/=temp;
if(a[i]%2)sum1++;
sum+=a[i];
}
if(sum%2){
cout<<'0';
return 0;
}
ll m=sum/2;
for(ll i=1;i<=n;i++){
for(ll j=m;j>=a[i];j--)dp[j]=max(dp[j],dp[j-a[i]]+a[i]);}
if(dp[m]==m){
puts("1");
pre(1,n){
if(a[i]%2){cout<<i;
return 0;}
}
}
else cout<<"0";
return 0;
}