NYOJ - 325 - zb的生日(dfs减枝,)

描述

今天是阴历七月初五，acm队员zb的生日。zb正在和C小加、never在武汉集训。他想给这两位兄弟买点什么庆祝生日，经过调查，zb发现C小加和never都很喜欢吃西瓜，而且一吃就是一堆的那种，zb立刻下定决心买了一堆西瓜。当他准备把西瓜送给C小加和never的时候，遇到了一个难题，never和C小加不在一块住，只能把西瓜分成两堆给他们，为了对每个人都公平，他想让两堆的重量之差最小。每个西瓜的重量已知，你能帮帮他么？

输入

多组测试数据（<=1500）。数据以EOF结尾
第一行输入西瓜数量N (1 ≤ N ≤ 20)
第二行有N个数，W1, …, Wn (1 ≤ Wi ≤ 10000)分别代表每个西瓜的重量

输出

输出分成两堆后的质量差

样例输入

5
5 8 13 27 14

样例输出

3

题目意思是这样的，有n个西瓜，把这n个西瓜分成两堆，求出这两堆质量的最小差。

思路：求数列的子集，并且子集的和不超过所有数的和的一半，如果超过就跳出，算出差值，如果差值比当前最小差值要小，那么把最小差值更新为当前差值。继续递归，寻找是否有更小的情况，知道递归完成。

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n,sum1,sum2,ans;
int a[25];
int vis[10000];

void dfs(int cur){
	
	if(sum2>=(sum1+1)/2){//递归边界 
		//如果找到0的情况，那么停止搜索，因为没有比0还小的 
		if(sum1%2==0&&sum1/2==sum2){	
			ans = 0;
			return;
		}else{
			if(ans>2*sum2-sum1){
				ans = 2*sum2-sum1;
			}
		}
	}
	for(int i=cur ;i<n ;i++){
		if(!vis[i]){
			sum2 += a[i];
			vis[i] = 1;//设置访问标记 
			dfs(i+1);
			sum2 -= a[i];
			vis[i] = 0;//清楚标记 
		}
	}
}

int main(){
	while(scanf("%d",&n)!=EOF){
		memset(vis,0,sizeof(vis));
		//ans一定要足够大， 
		sum1 = 0; sum2 = 0; ans = 1000000;
		for(int i=0 ;i<n ;i++){
			scanf("%d",&a[i]);
			sum1 += a[i];
		}

		dfs(0);
		printf("%d\n",ans);
	}
	
	return 0;
}