Poj Sticks(木棒) DFS+剪枝

描述

乔治拿来一组等长的木棒，将它们随机地裁断，使得每一节木棍的长度都不超过50个长度单位。然后他又想把这些木棍恢复到为裁截前的状态，但忘记了初始时有多少木棒以及木棒的初始长度。请你设计一个程序，帮助乔治计算木棒的可能最小长度。每一节木棍的长度都用大于零的整数表示。

输入

输入包含多组数据，每组数据包括两行。第一行是一个不超过64的整数，表示砍断之后共有多少节木棍。第二行是截断以后，所得到的各节木棍的长度。在最后一组数据之后，是一个零。

输出

为每组数据，分别输出原始木棒的可能最小长度，每组数据占一行。

样例输入

9

5 2 1 5 2 1 5 2 1

4

1 2 3 4

0

样例输出

6

5

解题思路：DFS+剪枝，搜索的终点有多种，我选择了用完所有的小木段为终止条件。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#define MAX 70
int n;
int sticks[MAX];
int sum_len;

int cmp(const void *a,const void *b)
{
    return *((int *)b)-*((int *)a);
}

bool dfs(bool *flag,int len,int a_len,int s,int num)        //flag标记，len当前长度，a_len目标单根长度，s当前搜索位置，num当前总用量
{
    if(num==n)
        return true;
    int p=-1;                                                                                       //高效剪枝，不递归相同且不可用的枝条
    for(int i=s;i<n;i++){
        if(flag[i] || p==sticks[i]) continue;
        flag[i]=1;
        if(len+sticks[i]<a_len){
            if(dfs(flag,len+sticks[i],a_len,i,num+1)) return true;              //注意不成立还需要继续下一个小木段
            else
                p=sticks[i];
        }
        else if(len+sticks[i]==a_len){
            if(dfs(flag,0,a_len,0,num+1))return true;
            else
                p=sticks[i];
        }
        flag[i]=0;

        if(len==0)<span style="white-space:pre">	//剪枝，当没有能够组合出木棍，说明木棍可能已经用完了。
            break;
    }
    return false;
}

int main()
{
    bool flag[MAX];
    while(~scanf("%d",&n) && n){
        sum_len=0;
        for(int i=0;i<n;i++){
            scanf("%d",&sticks[i]);
            sum_len+=sticks[i];
        }
        qsort(sticks,n,sizeof(int),cmp);

        int MinLen=sum_len;
        for(int i=sticks[0];i<=sum_len;i++){
            if(sum_len%i==0){
                memset(flag,0,sizeof(flag));<span style="white-space:pre">			//这句感觉多余，为了安心还是放着吧
                if(dfs(flag,0,i,0,0)){
                    MinLen=i;
                    break;
                }
            }
        }
        printf("%d\n",MinLen);
    }
}