首页 > 代码库 > POJ-1011-Sticks

POJ-1011-Sticks

题目链接:http://poj.org/problem?id=1011

这道题用到了深搜+剪枝。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int a[65];
int vis[65];
int n;
int cmp(int x,int y)
{
    return x>y;
}
int dfs(int len,int need,int left)//len是原棒长度,need是拼成一个原棒长还需要的棒长,left是还剩下的棒子数
{
    int i;
    if(need==0&&left==0)return len;//当全部的棒子都正好拼完,就返回原棒长的数值
    if(need==0)need=len;//拼完一个原棒长,继续拼下一个
    for(i=0;i<n;i++)
    {
            if(vis[i])continue;
            if(a[i]<=need)
            {
                vis[i]=1;
                if(dfs(len,need-a[i],left-1))return len;
                vis[i]=0;
                if(len==need||need==a[i])break;//剪枝:子情况不能拼成原棒长,直接跳出循环
                while(a[i]==a[i+1])i++;//剪枝:如果前面的棒子拼不成原棒长,后面相同长度的棒子直接掠过,因为肯定也拼不成。
            }
    }
    return 0;
}
int main()
{
    int i;
    while(cin>>n&&n)
    {
        int flag,b,c,d,sum=0;
        for(i=0;i<n;i++)
        {
            cin>>a[i];
            sum+=a[i];
        }

        sort(a,a+n,cmp);
        for(i=a[0];i<=sum;i++)
        {
            memset(vis,0,sizeof(vis));
            if(sum%i==0)//剪枝:棍棒总长度必须是原棒成的倍数
            {
                flag=dfs(i,i,n);
                if(flag)
                {
                    break;
                }

            }
        }
        cout<<flag<<d<<endl;
    }
    return 0;
}