木棒(dfs+各种剪枝方法)

题目:

见https://www.acwing.com/problem/content/169/

思路:

从当前木棍中挑出一根最长木棍,从该长度开始枚举所有长度,dfs拼木棒…
剪枝:
详情见《算法进阶竞赛指南》P105

代码:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=70;
int sticks[N],n,sum,length;//sum是所有木棍长度总和,length是可行的木棒长度
bool visit[N];

bool dfs(int u,int cur,int start)//u是已经拼好的木棒数量,cur是当前正在拼的木棒长度,start是为了避免出现1 3 2和3 2 1这种情况,记录上次使用到的木棒的位置
{
    if(sum==u*length) return true;//如果已经拼好的木棒数量等于总长度即可返回
    if(cur==length) return dfs(u+1,0,0);//如果当前已经拼好了一个木棒,那么dfs进入下一层
    for(int i=start;i<=n;i++)
    {
        if(visit[i]) continue;//如果该木棒拼过了,就跳过该木棒
        int l=sticks[i];
        if(cur+l>length) continue;//如果该木棍加上此时l大于length那么也跳过该木棒
        visit[i]=1;
        if(dfs(u,cur+l,i+1)) return true;//选择当前木棍递归进入下一层
        visit[i]=0;
        if(!cur) return false;//三个剪枝,详见书
        if(cur+l==length) return false;
        int j=i+1;
        while(sticks[i]==sticks[j]) j++;
        i=j-1;
    }
    return false;
}
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    while(cin>>n,n)
    {
        memset(visit,0,sizeof visit);
        length=0;
        sum=0;
        for(int i=1;i<=n;i++)
    {
        int l;
        cin>>l;
        sticks[i]=l;
        sum+=l;
        length=max(length,l);
    }
    sort(sticks+1,sticks+n+1,cmp);
    while(length)
    {
        if(sum%length==0&&dfs(0,0,0))//木棍总长度一定可以整除length
        {
            cout<<length<<endl;
            break;//如果找到了一个可行方案,那么该length就是最优解
        }
        length++;
    }
    }
    return 0;
}