描述
乔治拿来一组等长的木棒,将它们随机地裁断,使得每一节木棍的长度都不超过50个长度单位。然后他又想把这些木棍恢复到为裁截前的状态,但忘记了初始时有多少木棒以及木棒的初始长度。请你设计一个程序,帮助乔治计算木棒的可能最小长度。每一节木棍的长度都用大于零的整数表示。
输入
输入包含多组数据,每组数据包括两行。第一行是一个不超过64的整数,表示砍断之后共有多少节木棍。第二行是截断以后,所得到的各节木棍的长度。在最后一组数据之后,是一个零。
输出
为每组数据,分别输出原始木棒的可能最小长度,每组数据占一行。
样例输入
9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0
样例输出
6
5
代码
#include<bits/stdc++.h>
using namespace std;
int n,sum,num;
bool flag;
int a[70],vis[70];
void dfs(int x,int pid,int cnt)
{
if(flag) return;
if(x==pid) cnt++,x=0;
if(cnt==sum/pid)
{
flag=true;
return;
}
for(int k=n-1;k>-1;k--)
{
if(x+a[k]>pid||vis[k]) continue;
if(a[k]==a[k-1]&&!vis[k-1]) continue;
vis[k]=1;
dfs(x+a[k],pid,cnt);
vis[k]=0;
if(flag) return;
if(x==0) return;
}
}
int main()
{
while(cin>>n)
{
if(n==0) break;
memset(a,0,sizeof(a));
sum=0;num=0;
for(int i=1;i<=n;i++)
{
int tmp;
scanf("%d",&tmp);
if(tmp<=50)
{
a[num++]=tmp;
sum+=tmp;
}
}
sort(a,a+n);
for(int i=a[n-1];i<=sum;i++)
{
if(sum%i!=0) continue;
flag=false;
dfs(0,i,1);
if(flag)
{
cout<<i<<endl;
break;
}
}
}
return 0;
}
一道搜索题目,重点在于如何利用深度搜索去配对木棒。其中,为了减少重复操作,对木棒长度进行排序,从最长的木棒开始进行搜索,越短的木棒灵活性越大,若所有木棒匹配成功,则返回true。
PS:一开始写的dfs返回值不能保证当匹配不成功时一定返回false,导致wa;后来由于是从短木棒开始搜索而导致超时。