[USACO 2012OPEN] Balanced Cow Subsets(meet in the middle + DFS)

题目大意：给出n(n<=20)个正整数，选出m个数，把这m个数分成两组，使得两组的和相等，求第一步选数的方法。注意：可能有一种选数方案存在多种划分方案，但是只算一种。

思路：设ai∈{-1,0,1}，所求的集合可以视作Σai*xi=0，说得简单一点就是（以n = 6为例）

a1*x1 + a2*x2 + a3*x3 + a4*x4 + a5*x5 + a6*x6 = 0

显然枚举6个x肯定超时。于是我们采用meet in the middle的思路，拿出一半分成一组。a1*x1 + a2*x2 + a3*x3记为leftsum，a4*x4 + a5*x5 + a6*x6记为rigthsum。枚举leftsum和rightsum的时间复杂度为O(3^(m/2))是原来时间复杂度O(3^m)的根号倍。用状态压缩记录每种和的情况，最后扫描记录的leftsum和rightsum相应的leftsum+rightsum=0就是一个符合要求的集合。

这里还有一个技巧，在执行枚举rightsum操作的时候我们就已经可以求出相应的leftsum，具体参考我的代码。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define MAXN 21
using namespace std;
int num[MAXN];
pair<int,int> res1[60005];//3^10 
int len_1;
bool vis[1<<20];
bool flag;

int mylower_bound(int a)
{
	int l = 1,r = len_1,mid,out = 1;
	while(l <= r)
	{
		mid = (l+r)/2;
		if(res1[mid].second == a)
		{
			out = mid;
			r = mid - 1;
		}
		else if(res1[mid].second < a) l = mid + 1;
		else r = mid - 1;
	}	
	return out;
}
int myup_bound(int a)
{
	int l = 1,r = len_1,mid,out = 0;
	while(l <= r)
	{
		mid = (l+r)/2;
		if(res1[mid].second == a)
		{
			out = mid;
			l = mid + 1;
		}
		else if(res1[mid].second < a) l = mid + 1;
		else r = mid - 1;
	}
	return out;
}

void dfs(int dep,int s,int sum,int limit)
{
	if(dep > limit) 
	{
		if(flag == 0) {len_1++;res1[len_1].first = s;res1[len_1].second = sum;return;}//求leftsum
		else//二分法求相应的和
		{
			int x = mylower_bound(-sum);//最前面的leftsum
			int y = myup_bound(-sum);//最后面的leftsum
			for(int i = x; i <= y; i++)
				vis[res1[i].first + s] = 1;
			return;
		}
	}
	for(int i = -1; i <= 1; i++)//a = -1,0,1
		dfs(dep + 1,s+(i != 0? (1<<(dep-1)) : 0),sum + i*num[dep],limit);
}

bool cmp(pair<int,int> a,pair<int,int> b)
{
	if(a.second != b.second)
		return a.second < b.second;
	else return a.first < b.first;
}

int main()
{
	int n;
	scanf("%d",&n);
	for(int i = 1; i <= n; i++) scanf("%d",&num[i]);

	dfs(1,0,0,n/2);
	sort(res1+1,res1+len_1+1,cmp);//和从小到大排序，便于查找相等的和
	flag = 1;
	dfs(n/2+1,0,0,n);

	int ans = 0;
	int len = 1<<n;
	for(int k = 1; k < len; k++)//查找
		if(vis[k]) ans++;
	printf("%d\n",ans);
}