1. 程式人生 > >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;
}