poj 1011 Sticks - 经典DFS

点击打开题目链接

//poj1011经典搜索
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a[65];
bool vis[65];
int n;

int dfs(int len,int need_len,int num)
/*三个系数分别代表 当前的搜索长度， 达到目的要求还需要的
长度， 还未拼接的木棒个数 */
{
if(need_len==0 && num==0)
//还需拼接的长度为 0 并且 木棒都被拼接
{
return len;
}
if(need_len==0)
{
need_len=len;
}
for(int i=0;i<n;i++)
{
if(vis[i]==1) //如果已经拼接过了
{
continue;
}
else
{
if(need_len>=a[i]) //符合要求吗？
{
vis[i]=1; //符合
if(dfs(len,need_len-a[i],num-1))//判断是否符合条件
return len;
//下面是不符合条件时的各种剪枝
vis[i]=0;//恢复状态
if(a[i]==need_len||need_len==len)
{
//表示拼接完成 或者 刚进行新一轮拼接
//但是 上一个if不符合，即 子问题不能完成所有任务
//不再考虑，直接跳出，搜索失败！
break;
}
while(a[i]==a[i+1])
//a[i]不能完成任务，与它相等长度的也完不成任务！
{
i++;
}
}
}
}
return 0;
}
bool cmp(int a,int b)
{
return a>b;
}

int main()
{
while(~scanf("%d",&n),n)
{
int sum=0;
memset(a,0,sizeof(a)) ;
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
sum+=a[i];
}
int k=0,len;
sort(a,a+n,cmp);//剪枝，大的数对后面的约束力更大
for(len=a[0];len<=sum;len++)//遍历出最小的树枝长度
{
memset(vis,0,sizeof(vis));
if(sum%len==0)
{
k=dfs(len,0,n);
if(k) break;
}
}
printf("%d\n",k);
}

return 0;
}