题目:Sticks
题意:
一大堆相同长度的木棍被随意切割成n份小木棍,
现想要再次拼接成一大堆相同长度的木棍,求原始最短木棍的可能长度.
思路:
从小到大依次枚举所有的可能长度len,第一次找到可拼接成的len就是答案.
当打算把所有小木棍拼接成一大堆长度为len木棍时,
需要木棍之间进行组合,不能确定谁和谁属于一根的,
因此需要用到dfs算法,当发现此刻拼接的不对时,可以进行回溯,返回到上一步.
剪枝(优化):
优化1:
——对小木棍进行从大到小排序.
原因:
在选择木棍拼接时,原则上只要不超过假定的长度len,就可拼接.
但不能盲目的去选择小木棍(一会选长,一会选短),会造成一些重复的搜索。
因此可以规定:总是先长后短。
看下图:
优化2:
——在截断后的排好序的木棍中,当用木棍i拼和len时,可以从第i+1后的木棍开始搜。
因为根据优化1,i 前面的小木棍已经搜过,再搜就会发生重复.
优化3:
——用当前最长长度的木棍开始搜,如果拼不出当前设定的len,可提前返回,换下一个len。
举例1:10 8 6 5 1
当遍历到8时,无论怎么拼接,都不能拼成len,这时候就可提前返回了,即换下一个len了。
执行代码部分:if(l+a[i]==len) return 0;
遍历到8时,发现8组不成len,这时候就会再次放回上一层(l=0),
接着执行代码部分,此时代码中的a[i]等于10。
举例2:8 6 5 1
当遍历到8时,无论怎么拼接,都不能拼成len,这时候就可提前返回了,即换下一个len了。
执行代码部分:if(l==0) return 0;
遍历到8时,发现8组不成len,这时候就会再次放回上一层(l=0),
接着执行代码部分。
二者可能有点重叠,但确实缺一不可,作者水平有限,只能理解到这。
优化4:
——相同长度的小木棍不要搜索多次。用当前长度的小木棍搜下去得不到结果时,用一支同样长度的还是得不到结果,也可提前返回到上一层.。
举例:4 3 3 1
若len=9
4与第1个3结合会发现不成功,那4和第2个3结合也没有意义,遇到第2个3时,直接跳过即可。
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[110],book[110];
int n,len,cnt;
bool cmp(int a,int b){
return a>b;
}
//num表示当前组装的第几根木棍.
//l表示当前组装的木棍的长度.
//x的作用:当前已拼接的木棍只与区间[x,n]的进行拼接.
int dfs(int num,int l,int x)
{
if(num==cnt+1) return 1;
if(l==len) return dfs(num+1,0,1);
int fail=0;
for(int i=x;i<=n;i++)//优化2
{
if(!book[i]&&l+a[i]<=len&&fail!=a[i])
{
book[i]=1;
if(dfs(num,l+a[i],i+1)==1)
return 1;
//若未执行return 1,就说明a[i]虽不违背拼接原则,但无法用于当前拼接.
fail=a[i];
book[i]=0;
if(l==0||l+a[i]==len)//优化3
return 0;
}
}
return 0;
}
int main()
{
while(~scanf("%d",&n)&&n)
{
int sum=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum+=a[i];
}
sort(a+1,a+1+n,cmp);//优化1
for(len=a[1];len<=sum;len++)
{
if(sum%len!=0) continue;
cnt=sum/len;
memset(book,0,sizeof(book));
if(dfs(1,0,1)==1)
break;
}
printf("%d\n",len);
}
return 0;
}