原文:http://user.qzone.qq.com/289065406/blog/1311647833
http://www.cnblogs.com/lyy289065406/archive/2011/07/31/2122544.html
poj经典的搜索题+剪枝
Initlen:题目要求的最短木棒的长度
sumlen:所有木棒的总和
maxlen:将所给的数组从大到小排序后的第一个数字,即最长的木棒
vist:木棒是否被搜索过
stick:存放木棒的数组
s:stick数组开始搜素的位置
num:已经搜索过得木棒数量
dfs:实现木棒的拼凑,用dfs求出所有的小棒能否拼凑成这个长度
就是比如说17,9,9,9,9,8,8,5,2……如果当前最长小棒为17,它与第一个9组合之后dfs发现不能拼成len,那么17就不用和后面所有的9组合了,而直接和8开始组合。
思路:题目所要求的最短木棒一定在[maxlen,sumlen]中, 若能在[maxlen,sumlen-InitLen]找到最短的InitLen,该InitLen必也是[maxlen,sumlen]的最短;若不能在[maxlen,sumlen-InitLen]找到最短的InitLen,则必有InitLen=sumlen;
int cmp(const void* a,const void* b)
{
return *(int*)b-*(int*)a;
}
*(int*)b的意思是将b转化成int*再取b的值。a-b和b-a分别是从小到大和从大到小,qsort是quick sort的简写,其实就是快速排序,所以每次就比较两个值 qsort的调用方法是 void qsort(void *base, int nelem, int width, int (*fcmp)(const void *,const void *)); base是指针入口(stick),nelem是排序的元素个数(n),width是元素大小(sizeof(stick)),comp(cmp)就是比较函数了。
剪枝主要有4个地方
1,如果要想Initlen符合要求(原先每根木棒长度相等),那么sumlen%Initlen==0
2,譬如有 23 17 17 17 17 8 7 6一组数据,如果23+17此时不符合,就不要重复计算后面的17
3,若能在[maxlen,sumlen-InitLen]找到最短的InitLen,该InitLen必也是[maxlen,sumlen]的最短输出即可;若不能在[maxlen,sumlen-InitLen]找到最短的InitLen,则必有InitLen=sumlen输出即可;
4,最重要的剪枝:对于某个目标InitLen,在每次构建新的长度为InitLen的原始棒时,检查新棒的第一根stick[i],若在搜索完所有stick[]后都无法组合,则说明stick[i]无法在当前组合方式下组合,不用往下搜索(往下搜索会令stick[i]被舍弃),直接返回上一层
#include<iostream>
#include<algorithm>
using namespace std;
/*Initlen:题目要求的问题;sumlen:所有木棒的长度的总和*/
int cmp(const void* a,const void* b)
{
return *(int*)b-*(int*)a;
}
int n;
bool dfs(int* stick,bool* vist,int len,int Initlen,int s,int num)
{
if(num==n)
{
return true;
}
int sample = -1;
int i;
for(i=s;i<n;i++)
{
if(vist[i] || stick[i]==sample)//2
{
continue;
}
vist[i] = true;
if(len+stick[i]<Initlen)
{
if(dfs(stick,vist,len+stick[i],Initlen,i,num+1))
return true;
else
{
sample = stick[i];
}
}
else if(len+stick[i]==Initlen)
{
if(dfs(stick,vist,0,Initlen,0,num+1))
return true;
else
sample=stick[i];
}
vist[i]=false;
if(len==0) //4
break;
}
return false;
}
int main(void)
{
while(cin>>n && n)
{
int* stick = new int[n];
bool* vist = new bool[n];
int sumlen = 0;
int i;
for(i=0;i<n;i++)
{
cin>>stick[i];
sumlen += stick[i];
vist[i] = false;
}
qsort(stick,n,sizeof(stick),cmp);
int maxlen = stick[0];
bool flag = false;
int Initlen;
for(Initlen=maxlen;Initlen<=sumlen-Initlen;Initlen++)//1
{
if(!(sumlen%Initlen)&& dfs(stick,vist,0,Initlen,0,0))
{
cout<<Initlen<<endl;
flag = true;
break;
}
}
if(!flag)
{
cout<<sumlen<<endl;
}
delete stick;//释放内存
delete vist;
}
return 0;
}