POJ1011 dfs剪枝

B站上老师已经讲的很清楚了,我挂个代码;

B站:https://www.bilibili.com/video/av10046345/?p=18

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const double epos=1e-8;
const int maxn=100;
int a[maxn];
int n;
int flag[maxn];
int L;
int lastst;
int dfs(int r,int m){
    if(r==0&&m==0) return 1;
    if(m==0) m=L;

    /*
    根子拼接中若出现前面的这一小部分比后面这一小部分长,则说明对调后,
    按照顺序也是可以拼接的,于是这个st可以保证拼接棍子中,小棍子都是有序的;
    */
    int st=(m==L?0:lastst+1);
    for(int i=st;i<n;i++){
        if(flag[i]==1||a[i]>m) continue;
        //上一根拼接不成功,则不必再试和它长度相同的根子;
        if(i>0&&flag[i-1]==0&&a[i]==a[i-1]) continue;
        flag[i]=1;
        lastst=i;
        if(dfs(r-1,m-a[i])) return 1;
        //情况不成立,回溯;
        flag[i]=0;
        //后面拼接不成功,回溯到了最开始,而替换第一根木棍是没有意义的,因为替换最长的这一根后,他仍会出现在后续棍子中;
        if(m==L) return 0;

        //这个剪枝我还不理解...太菜了QAQ;
        if(a[i]==m) return 0;
    }
    return 0;
}

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

int main(){

    while(scanf("%d",&n)!=EOF&&n){
        int sum=0;
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        int maxx=sum;
        sort(a,a+n,cmp);
        int hhh=0;
        for(L=a[0];L<=sum/2;L++){
            if(sum%L)
                continue;
            memset(flag,0,sizeof(flag));
            lastst=0;
            if(dfs(n,L)){
                hhh=1;
                break;
            }
        }
        if(hhh)
            printf("%d\n",L);
        else
            printf("%d\n",maxx);
    }


    return 0;
}

最后一个剪枝,先抄大佬一段话://剪枝3:假设len[i]为最后一根木棒,被更小木棒替换后最终能够成功,那么3必然出现在后面的某个棍子k里。 //将棍子k中的len[i]和棍子i中用来替换3的几根木棒对调,结果当然一样是成功的。这就和i原来的拼法会导致不成功矛盾。

这段话摘自:https://blog.csdn.net/tp7309/article/details/53447343

猜你喜欢

转载自blog.csdn.net/chenyume/article/details/84250844