UVA 1354 Mobile Computing 枚举二叉树 *

题目地址：http://vjudge.net/problem/UVALive-3403

要使天平平衡，满足n*a=m*b （总量为n，m），所以不管n，m多少，总可以找到两个a，b使天平平衡，所以，只需要递归的暴力枚举每个天平的左右两端的重量，保存其最大长度就好

那么怎么枚举重量呢，因为挂坠也就6个，所以可以用二进制表示，算出所有挂坠任意组合的子集

考虑数据结构：

1）枚举出所有二叉树，那怎么计算每棵树最大的L，最大R？ 用个Tree结构体保存该根的二叉树的L，R，即tree[i].L和tree[j].R表示i为根的树的L，R

2）怎么保存枚举出来的所有二叉树？用vector<Tree> tree[]  ,tree[i]保存子集为i的所有树的可能，（因为只需要知道L，R就够了，所以只保存这个）

#include <bits/stdc++.h>
using namespace std;
#define REP(i,a,b)  for(int i=a;i<=(int)(b);++i)
#define REPD(i,a,b) for(int i=a;i>=(int)(b);--i)
const int maxn=6;
struct Tree
{
	double L,R;
	Tree():L(0),R(0){};
};
double r,w[1<<maxn],sum[1<<maxn];
int vis[1<<maxn];
vector<vector<Tree> > tree(1<<maxn);
void DFS(int subset){
	if(vis[subset]) return ;
	vis[subset]=true;

	bool haveChild=false;
	for(int left=(subset-1)⊂left;left=(left-1)&subset){
		haveChild=true;
		int right=subset^left;
		DFS(left); DFS(right);

		double d1=sum[right]/sum[subset];
		double d2=sum[left]/sum[subset];

		REP(i,0,tree[left].size()-1) REP(j,0,tree[right].size()-1){
			Tree t;
			t.L=max(tree[left][i].L+d1,tree[right][j].L-d2);
			t.R=max(tree[left][i].R-d1,tree[right][j].R+d2);
			if(t.L+t.R<r) tree[subset].push_back(t);
		}
	}
	if(!haveChild) tree[subset].push_back(Tree());
}
int main(int argc, char const *argv[])
{
	int T; scanf("%d",&T);
	while(T--){
		int n; scanf("%lf%d",&r,&n);
		REP(i,0,n-1) scanf("%lf",&w[i]); 

		int root=(1<<n)-1;
		REP(i,0,root){
			sum[i]=0; 
			tree[i].clear();
			REP(j,0,n-1) if((1<<j)&i) sum[i]+=w[j];
		}
		memset(vis,0,sizeof(vis));
		DFS(root);
		double ans=-1;
		REP(i,0,tree[root].size()-1) ans=max(ans,tree[root][i].L+tree[root][i].R);
		printf("%.10lf\n", ans);
	}
	return 0;
}