自己加了些注释和排序方法
转载请注明出处:優YoU http://user.qzone.qq.com/289065406/blog/1311647833
解题思路:
DFS+剪枝
POJ2362的强化版,重点在于剪枝
令InitLen为所求的最短原始棒长,maxlen为给定的棒子堆中最长的棒子,sumlen为这堆棒子的长度之和,那么InitLen必定在范围[maxlen,sumlen]中
根据棒子的灵活度(棒子越长,灵活度越低) DFS前先对所有棒子降序排序
剪枝:
1、 由于所有原始棒子等长,那么必有sumlen%Initlen==0;
2、 若能在[maxlen,sumlen-InitLen]找到最短的InitLen,该InitLen必也是[maxlen,sumlen]的最短;若不能在[maxlen,sumlen-InitLen]找到最短的InitLen,则必有InitLen=sumlen;
3、 由于所有棒子已降序排序,在DFS时,若某根棒子不合适,则跳过其后面所有与它等长的棒子;
4、 最重要的剪枝:对于某个目标InitLen,在每次构建新的长度为InitLen的原始棒时,检查新棒的第一根stick[i],若在搜索完所有stick[]后都无法组合,则说明stick[i]无法在当前组合方式下组合,不用往下搜索(往下搜索会令stick[i]被舍弃),直接返回上一层
#include<iostream>
#include<algorithm>
using namespace std;
int cmp(const void* a,const void* b)
{
return *(int*)b-*(int*)a;
}
int cmp2(int a,int b)
{
return a>b;
}
int n; //木棒数量
bool dfs(int* stick,bool* vist,int len,int InitLen,int s,int num) //len:当前正在组合的棒长 InitLen:目标棒长
{ //s:stick[]的搜索起点 num:已用的棒子数量
if(num==n)return true;
int sample=-1;
//sample即发现当前用该长度的去拼凑是不会成功的,当有相等长度的出现时直接跳过.\
因为排序过的,所有可以只用一个变量标记很多个相等的stick[]
for(int i=s;i<n;i++)
{
if(vist[i] || stick[i]==sample) //剪枝3,等长的木棒只搜索一次
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;
for(int i=0;i<n;i++)
{
cin>>stick[i];
sumlen+=stick[i];
vist[i]=false;
}
//qsort(stick,n,sizeof(stick),cmp);
sort(stick,stick+n,cmp2);
//sort(stick,stick+n);
//reverse(stick,stick+n);
int maxlen=stick[0]; //最大的棒为InitLen的搜索起点
bool flag=false;
//剪枝1,若能在[maxlen,sumlen-InitLen]找到最短的InitLen,该InitLen必也是[maxlen,sumlen]的最短
for(int InitLen=maxlen;InitLen<=sumlen-InitLen;InitLen++) //InitLen:原始棒长
{ //剪枝2,InitLen必是sumlen的约数(InitLen<=sumlen-InitLen:意思不就是至少有两个木棒)
if(!(sumlen%InitLen) && dfs(stick,vist,0,InitLen,0,0))
{
cout<<InitLen<<endl;
flag=true;
break;
}
}
if(!flag)printf("%d\n",sumlen);
delete stick;
delete vist;
}
return 0;
}