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_backconst 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;}