POJ 1011 搜索剪枝

题意:给一堆木棒,这些木棒是由几根长度相同的长木棒厥断而成的,求最短的长木棒长度能厥成这些短的木棒。

题解:

   所求长度肯定介于最长木棒和木棒之和中间,并且可以整除木棒之和。于是穷举所有可能的值,选中一个值之后,用dfs给木棒寻找组合,能找到就输出。

  直接dfs会超时,所以需要剪枝。可以先排序,这样相同的木棒就聚在了一起,对于相同的木棒,只要这一长度的第一个木棒没被用,那么这些木棒都一定是不可以的,这样就可以过了。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int a[100],vis[100];
int n;
int len_max,len_sum,len;
bool flag;
int cmp(int a,int b){
	return a>b;
}
void dfs(int deep,int l,int num )//deep深度,也就是用了几个棒 l当前长度 num搜到的位置
{
	if(flag)return ;
	if(l==0)
	{
		int k=0;
		while(vis[k])k++;
		vis[k]=1;
		dfs(deep+1,a[k],k+1);
		vis[k]=0;//回溯
		return ;
	}
	if(l==len )
	{
		if(deep==n){
			flag=true;
			return ;
		}
		dfs(deep,0,0);
		return ;
	}
	int i;
	for(i=num;i<n;i++)
	{
		if(!vis[i]&&l+a[i]<=len){
			if(!vis[i-1]&&a[i]==a[i-1])continue;//上一个一样的用不上,这个也一定用不上
			vis[i]=1;
			dfs(deep+1,l+a[i],i+1);
			vis[i]=0;
		}
	}
}
int main()
{
	int i;
	while(~scanf("%d",&n)){
		flag=0;
		if(!n)break;
		len_max=len_sum=0;
		for(i=0;i<n;i++)scanf("%d",&a[i]);
		sort(a,a+n,cmp);
		len_max=a[0];
		for(i=0;i<n;i++)len_sum+=a[i];
		for(len=len_max;len<=len_sum;len++){
			if(len_sum%len)continue;
			memset(vis,0,sizeof(vis));
			dfs(0,0,0);
			if(flag)break;
		}
		printf("%d\n",len);
	}
}

猜你喜欢

转载自blog.csdn.net/nwpu2017300135/article/details/81062182