C. Baby Ehab Partitions Again
好题!
记元素之和为sum
1.当sum是奇数时,不可能对半分,答案是0。
2.当sum是偶数时:
(1)当sum/2是一个奇数,且原序列所有元素是偶数时,答案是0(因为不可能用一些偶数拼凑出一个奇数)。
(2)否则,我们用dp判断一些sum/2是否可能被凑出来(时间复杂度
O
(
N
∗
s
u
m
)
O(N*sum)
O(N∗sum)),若不可能,答案是0,若可能,答案是1。
证明:对于序列
a
1
,
a
2
,
.
.
.
,
a
n
a_1,a_2,...,a_n
a1,a2,...,an,取gcd出来(记为d),那么可以把a序列表示为:
d
∗
(
b
1
,
b
2
,
.
.
.
,
b
n
)
d*(b_1,b_2,...,b_n)
d∗(b1,b2,...,bn),则b序列必定有一个数是奇数,因为若b序列没有奇数,应该可以再提一个因子2出来。又因为a序列是可以等分的,那么b序列也是可以等分的,即b序列的元素之和是偶数。这时候我们删去b序列中的一个奇数,那么b序列的元素之和是奇数,此时b序列无法再等分为两个子序列,即a序列无法等分。
#include <bits/stdc++.h>
using namespace std;
#define ull unsigned long long
#define mid (l+r>>1)
#define lson (rt<<1)
#define rson (rt<<1|1)
#define pb push_back
#define fi first
#define se second
#define int long long
#define pii pair<int,int>
const double PI=acos(-1.0);
const double eps=1e-6;
const int mod=1e7+7;
const int N=1e2+2;
const int M=2e5+5;
const int inf=0x3f3f3f3f;
int T,n,a[N],sum;
bool dp[M];
bool judge(){
dp[0]=1;
for(int i=1;i<=n;++i)
for(int j=sum-a[i];j>=0;--j) {
if(!dp[j+a[i]]) dp[j+a[i]]=dp[j];
}
return dp[sum/2];
}
signed main(){
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
// cin>>T;
T=1;
while(T--){
cin>>n;
bool ok=false;
for(int i=1;i<=n;++i) {
cin>>a[i];
sum+=a[i];
if(a[i]&1) ok=true; //有奇数
}
if((sum&1) || (!ok && (sum/2%2==1))) cout<<0<<'\n';
else {
if(!judge()) cout<<0<<'\n';
else {
cout<<1<<'\n';
int d=a[1];
for(int i=2;i<=n;++i) d=__gcd(d,a[i]);
for(int i=1;i<=n;++i){
if(a[i]/d%2==1){
cout<<i<<'\n';
break;
}
}
}
}
}
}