CDOJ 215 吴队长征婚 DFS+剪枝

题目链接:

http://acm.uestc.edu.cn/#/problem/show/215

题意:

题解:

http://www.lai18.com/content/7765203.html
剪枝1：大木棍长度只可能是sum的约数
剪枝2：重复的木棍就可以不用枚举了
剪枝3：每次都从最长的开始枚举

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 1e5+10;

int n,a[70],sum,num,vis[70];

bool cmp(int x,int y){
    return x > y;
}

bool dfs(int x,int y,int z,int len){
    if(x == num){
        return 1;
    }

    for(int i=z; i<n; i++){
        if(vis[i]) continue;
        // cout << x << " " << i << " " << y+a[i] << endl;
        if(y+a[i] == len){
            vis[i] = 1;
            if(dfs(x+1,0,0,len)) return 1; // 如果当前搜出了一个符合长度的，那么它肯定是一个合法的情况，就不需要拆它了
            vis[i] = 0;
            return 0; //如果当前这个木棒在合成后正好符合长度要求，但是之后的搜索确是失败的，说明后面也无法合成满足要求的木棒，也直接return false就可以
        }else if(y+a[i] < len){
            vis[i] = 1;
            if(dfs(x,y+a[i],i+1,len)) return 1; // 如果当前搜出了一个符合长度的，那么它肯定是一个合法的情况，就不需要拆它了
            vis[i] = 0;
            if(y==0) return 0; // 从长度为0开始dfs 最后回到了0，在这一层没有满足的，回溯
            while(a[i]==a[i+1]) i++; // 如果某个木棒在这个状态搜索失败，那所有与这个木棒相同长度的木棒就不用在这次搜索中搜索了
        }
    }
    return 0;
}

int main(){
    while(cin >> n, n){
        sum = 0;
        for(int i=0; i<n; i++){
            a[i] = read();
            sum += a[i];
        }
        sort(a,a+n,cmp);
        int len;
        for(len=a[0]; len<=sum; len++){ // 原长度肯定>=这些木棒的最大长度，而且能被所有木棒的总长度整除
            if(sum%len == 0){
                num = sum/len;
                // cout << "===" << num << endl;
                MS(vis);
                if(dfs(0,0,0,len))
                    break;
            }
        }
        cout << len << endl;
    }

    return 0;
}