【DFS剪枝】poj 1011 Sticks

最新推荐文章于 2021-02-01 23:21:52 发布

原创最新推荐文章于 2021-02-01 23:21:52 发布 · 349 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

搜索

本文详细解析了POJ1011 Sticks题目，介绍了一种通过深度优先搜索（DFS）来寻找原树枝最短可能长度的方法，并提供了完整的AC代码实现。文章还介绍了几种有效的剪枝策略以提高搜索效率。

poj 1011 Sticks

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

题意：

将未知根数的相同长度的树枝随机截断，给出截断后的树枝，求原树枝最短可能的长度。

思路：

将枚举可能的长度，dfs判断长度的可行性。

剪枝：将树枝从大到小排列，

1.如果长度大的树枝无法组合成（需判断的长度），那么不需继续判断。

2.在前面有完成组合，剩下的组合无法符合时，不需继续判断。

3.在找组合时，如果无法与前一个树枝组成一个组合，那么与前一个相同长度的树枝跳过判断。

ac代码：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <iostream>
using namespace std;
const int N = 101;

bool cmp(int a, int b)
{
    return a > b;
}
int n;
int a[N],v[N];
int sum,res,len;
bool dfs(int kai,int need, int all)
{
    if(all == 0)
        return 1;
    for(int i = kai; i < n; i++)
    {
        if(!v[i] && a[i] <= need)
        {
            v[i] = 1;
            if(a[i] == need)
            {
                if(dfs(0,len,all-a[i]))
                    return 1;
            }
            else
            {
                if(dfs(i+1,need-a[i],all-a[i]))
                    return 1;
            }
            v[i] = 0;
            if(all == sum) return 0;
            //剪枝一
            if(need == len) return 0;
            //剪枝二
            while(a[i+1] == a[i])
                i++;
            //剪枝三
        }
    }
    return 0;
}
int main()
{
    while(~scanf("%d",&n),n)
    {
        sum = 0;
        for(int i = 0; i < n; i++)
        {
            scanf("%d",&a[i]);
            sum += a[i];
        }
        sort(a,a+n,cmp);
        len = a[0];
        res = sum;
        memset(v,0,sizeof(v));
        for(; len < sum; len++)
        {
            if(sum%len == 0)
            {
                if(dfs(0,len,sum))
                {
                    res = len;
                    break;
                }
            }
        }
        printf("%d\n",res);
    }
    return 0;
}