Sticks dfs和剪枝

乔治拿来一组等长的木棒,将它们随机的砍掉,得到若干根小木棍,并使每一节小棍的长度都不超过50个单位。然后他又想把这些木棍拼接起来,恢复到裁剪前的状态,但他忘记了初始时有多少木棒以及木棒的初始长度。请你设计一个程序,帮助乔治计算木棒的可能最小长度,每一节木棍的长度都用大于零的整数表示。
输入包含多组数据,每组数据包括两行。第一行是一个不超过64的整数,表示砍断之后具有多少节木棍。第二行是截断以后,所得到的各节木棍的长度。在最后一组数据之后。是一个零。
对于每组数据,分别输出原始木棒的可能最小长度。
输入输出样例
输入
9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0
输出
6
5
将木棍从大到小排序,木棒的长度一定是所有木棍和的因数。
小木棍

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[1001],len,cnt;
int book[1001],n;
bool cmp(int a,int b)
{
    
    
    return a>b;
}
int dfs(int l,int k,int m)//剩余的长度,第几根,几组
{
    
    
    if(m==cnt)
        return 1;
    if(l==0)
        if(dfs(len,0,m+1))
            return 1;
    int i;
    for(i=k;i<n; i++)
    {
    
    
        if(book[i]==0&&a[i]<=l)
        {
    
    
            book[i]=1;
            if(dfs(l-a[i],i+1,m))
            return 1;
            book[i]=0;
            if(l==a[i]||l==len)
                break;
  /*若l==a[i],说明剩余的长度和当前木棍长度一样,
  但却不能和剩余的木棍组合在一起完成拼接,说明这木棍会被剩余,
  则之前已经组合好的木棍要重新组合。*/
  /*若l==len,说明利用当前木棍拼接时没有完成拼接,
  则这个木棍会被剩下,则之前已经组合好的木棍要重新组合。*/
            int tmp=a[i];
            while(tmp==a[i]&&i<n)
                ++i;
        }
    }
    return 0;
}
int main()
{
    
    
    int i,sum=0;
    scanf("%d",&n);
    for(i=0; i<n; i++)
    {
    
    
        scanf("%d",&a[i]);
        sum+=a[i];
    }
    sort(a,a+n,cmp);
    for(len=a[0]; len<sum; len++)
    {
    
    
        if(sum%len==0)
        {
    
    
            cnt=sum/len;
            if(dfs(len,0,0))
                break;
        }
    }
    printf("%d\n",len);
}

猜你喜欢

转载自blog.csdn.net/m0_46312382/article/details/107570862
今日推荐