BZOJ2679 : [Usaco2012 Open]Balanced Cow Subsets

考虑折半搜索，每个数的系数只能是-1，0，1之中的一个，因此可以先通过$O(3^\frac{n}{2})$的搜索分别搜索出两边每个状态的和以及数字的选择情况。

然后将后一半的状态按照和排序，$O(2^\frac{n}{2})$枚举前一半的每一个选择情况的状态，将该选择情况下所有状态按和排序，然后通过双指针求出所有合法状态。

时间复杂度$O(6^\frac{n}{2})$。

#include<cstdio>
#include<algorithm>
const int N=20,M=1<<10,E=59100;
int n,n0,i,j,k,a
,g[M],v[E],nxt[E],ed,m,q[M],ce,ans;bool vis[1<<N];
struct P{int s,S;}e[E];
inline bool cmp(const P&a,const P&b){return a.s<b.s;}
void dfsl(int x,int s,int S){
if(x==n0){v[++ed]=s;nxt[ed]=g[S];g[S]=ed;return;}
dfsl(x+1,s,S);
dfsl(x+1,s+a[x],S|(1<<x));
dfsl(x+1,s-a[x],S|(1<<x));
}
void dfsr(int x,int s,int S){
if(x==n){e[ce].s=s;e[ce++].S=S;return;}
dfsr(x+1,s,S);
dfsr(x+1,s+a[x],S|(1<<x));
dfsr(x+1,s-a[x],S|(1<<x));
}
int main(){
scanf("%d",&n);n0=(n+1)/2;
for(i=0;i<n;i++)scanf("%d",&a[i]);
dfsl(0,0,0),dfsr(n0,0,0);
std::sort(e+1,e+ce+1,cmp);
for(i=0;i<1<<n0;i++){
for(m=0,j=g[i];j;j=nxt[j])q[m++]=v[j];
std::sort(q,q+m);
for(j=k=0;j<ce;j++){
while(k<m&&q[k]<e[j].s)k++;
if(k==m)break;
if(q[k]==e[j].s)vis[i|e[j].S]=1;
}
}
for(i=1;i<1<<n;i++)if(vis[i])ans++;
return printf("%d",ans),0;
}