poj 1011 Stick(深度优先搜索)

57 篇文章 0 订阅

实实在在写不下去了,,真他妈的坑,一直超时……,下面是看的某位大神的代码

#include<iostream>  
#include<algorithm>  
using namespace std;  

int cmp(const void* a,const void* b)  
{  
    return *(int*)b-*(int*)a;  
}  

int n;  //木棒数量  

bool dfs(int* stick,bool* vist,int len,int InitLen,int s,int num)  //len:当前正在组合的棒长  InitLen:目标棒长    
{                                                                  //s:stick[]的搜索起点  num:已用的棒子数量  
    if(num==n)  
        return true;  

    int sample=-1;  
    for(int i=s;i<n;i++)  
    {  
        if(vist[i] || stick[i]==sample)  //剪枝3,等长的木棒只搜索一次  
            continue;  

        vist[i]=true;  
        if(len+stick[i]<InitLen)  
        {  
            if(dfs(stick,vist,len+stick[i],InitLen,i,num+1))  
                return true;  
            else  
                sample=stick[i];  
        }  
        else if(len+stick[i]==InitLen)  
        {  
            if(dfs(stick,vist,0,InitLen,0,num+1))  
                return true;  
            else  
                sample=stick[i];  
        }  
        vist[i]=false;  

        if(len==0)  //剪枝4,构建新棒时,对于新棒的第一根棒子,在搜索完所有棒子后都无法组合  
            break;  //则说明该棒子无法在当前组合方式下组合,不用往下搜索(往下搜索会令该棒子被舍弃),直接返回上一层  
    }  
    return false;  
}  

int main(void)  
{  
    while(cin>>n && n)  
    {  
        int* stick=new int[n];  
        bool* vist=new bool[n];  
        int sumlen=0;  

        for(int i=0;i<n;i++)  
        {  
            cin>>stick[i];  
            sumlen+=stick[i];  
            vist[i]=false;  
        }  

        qsort(stick,n,sizeof(stick),cmp);  
        int maxlen=stick[0];   //最大的棒为InitLen的搜索起点  

        bool flag=false;  
                             //剪枝1,若能在[maxlen,sumlen-InitLen]找到最短的InitLen,该InitLen必也是[maxlen,sumlen]的最短  
        for(int InitLen=maxlen;InitLen<=sumlen-InitLen;InitLen++)  //InitLen:原始棒长  
        {   //剪枝2,InitLen必是sumlen的约数  
            if(!(sumlen%InitLen) && dfs(stick,vist,0,InitLen,0,0))  
            {  
                cout<<InitLen<<endl;  
                flag=true;  
                break;  
            }  
        }  
        if(!flag)  
            cout<<sumlen<<endl;  

        delete stick;  
        delete vist;  
    }  
    return 0;  
}  
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值