木棒
乔治拿来一组等长的木棒,将它们随机地砍断,使得每一节木棍的长度都不超过50个长度单位。
然后他又想把这些木棍恢复到为裁截前的状态,但忘记了初始时有多少木棒以及木棒的初始长度。
请你设计一个程序,帮助乔治计算木棒的可能最小长度。
每一节木棍的长度都用大于零的整数表示。
输入格式
输入包含多组数据,每组数据包括两行。
第一行是一个不超过64的整数,表示砍断之后共有多少节木棍。
第二行是截断以后,所得到的各节木棍的长度。
在最后一组数据之后,是一个零。
输出格式
为每组数据,分别输出原始木棒的可能最小长度,每组数据占一行。
数据范围
数据保证每一节木棍的长度均不大于50。
输入样例:
9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0
输出样例:
6
5
本题整体的思想时dfs搜索,注意的几个点去剪枝。
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=70;
int sticks[N];
int sum,length;
int n;
bool st[N];
bool dfs(int u,int cur,int start)
{//传入参数为已经形成了几根棍子,当前棍子的长度,当前要遍历开始的位置
if(u*length==sum)return true;
if(cur==length)return dfs(u+1,0,0);
for(int i=start;i<n;i++)
{
if(st[i])continue;
int l=sticks[i];
if(cur+l<=length)
{
st[i]=true;
if(dfs(u,cur+l,i+1))return true;//只要形成的长度还没超过length,遍历位置就向后移
st[i]=false;
if(!cur)return false;
//该语句剪枝,当前棍子遍历了后面的所有情况,都没有形成可行性方案
//说明当前棍子的形成方案中的一些元素被前面的棍子使用了,所以当前方案之后不会有可行性方案
if(cur+l==length) return false;
//当前位置剪枝
//对应的是上一句剪枝,上一句剪枝会导致送从上一层循环的遍历发生变化
int j=i;//元素值相同遍历时跳过,剪枝
while(j<n&&sticks[j]==l)j++;
i=j-1;
}
}
return false;
}
int main()
{
while(cin>>n,n)
{
sum=0,length=0;
memset(st,false,sizeof st);
for(int i=0;i<n;i++)
{
cin>>sticks[i];
sum+=sticks[i];
length=max(length,sticks[i]);
}
sort(sticks,sticks+n);
reverse(sticks,sticks+n);//数倒序排列让遍历结果少的先接受遍历
//进行剪枝
while(true)
{
if(sum%length==0&&dfs(0,0,0))
{//当前长度可以被均分,并可分配到每一根棍子上
cout<<length<<endl;
break;
}
length++;
}
}
return 0;
}