POJ1011 Sticks

POJ1011 Sticks

剪枝好题，具体见代码：

 

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cctype>
#include <iostream>
using namespace std;

inline int read()
{
    int f=1,x=0; char ch;
    while(!isdigit(ch=getchar())) if(ch=='-') f=-1;
    while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    return f*x;
}

#define res register int
const int N=100;
int a[N],used[N],n,sum,m,ok,len;
bool cmp(const int &x,const int &y){return x>y;}

void dfs(int now,int last,int rest)
{
    if(rest==0)
    {    
        int i;
        if(now==m) {ok=1; return ;}
        for(i=1 ; i<=n ; i++) if(!used[i]) break;
        used[i]=1; dfs(now+1,i,len-a[i]);
        used[i]=0; if(ok) return ;
    }
    
    for(res i=last+1 ; i<=n ; i++)
    {
        if(!used[i]&&a[i]<=rest)
        {
            used[i]=1;    dfs(now,i,rest-a[i]);    used[i]=0;
            if(ok)    return ;
            while(a[i+1]==a[i]) i++;//
            if(rest==len||rest==a[i]) return ; 
        }
        if(i==n) return ; 
    }
}

/*剪枝：
1.把小木棍从大到小排序，先拼大的
2.从小到大枚举原始木块长度len,成立了就直接输出答案。范围：最短的木棍到所有木棍
长度之和的一半，且len需要整除sum
3.当一个木棍不能拼时，和它长度相等的木棍显然也不能拼
4.枚举下一个要拼的木棍时只需从上一个用过的木块的下一个开始枚举
****5.如果当前长棍剩余的未拼长度等于当前正在试着拼的木块的长度或者等于原始长度（即
刚开始拼），继续拼下去却失败了，则直接回溯，结束枚举。 
*/ 
int main()
{

    while(scanf("%d",&n)==1 && n)
    {
        sum=0;ok=0;
        memset(used,0,sizeof(used));
        memset(a,0,sizeof(a));
        for(res i=1 ; i<=n ; i++)
        {
            a[i]=read();
            sum+=a[i];
        }
        sort(a+1,a+n+1,cmp);
        for(len=a[1] ; len<=sum/2+1 ; len++)
        {
            if(sum%len!=0) continue;
             used[1]=1; m=sum/len;
            ok=0;
             dfs(1,1,len-a[1]);
             used[1]=0;
            if(ok) 
            {
                    printf("%d\n",len);
                    break;
            }
        }
        if(!ok) printf("%d\n",sum);
    }
    return 0;
}