算法学习之路|Poj1011Sticks（搜索剪枝）

最新推荐文章于 2022-10-15 14:19:48 发布

weixin_34306593

最新推荐文章于 2022-10-15 14:19:48 发布

阅读量141

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：https://yq.aliyun.com/articles/490103

本文介绍了一个使用深度优先搜索（DFS）算法解决的问题：给定一系列由原始棍子切割而成的小棍子，如何找出这些原始棍子可能的最小长度。通过排序和剪枝策略来提高算法效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：给n个棍子，这些棍子是从一些长度相等的原始棍子切割而来，问原始棍子的长度可能的最小值是多少。

跟着大佬的代码手敲了一遍，感觉离掌握dfs还有十万八千里。。。

剪枝的内容也挺重要：

木棍从大到小排列

重复长度的木棍跳过

若某一木棍是组成初始木棍的第一根，但之后匹配失败，说明此种初始长度无效（它必然是初始木棍的一部分，匹配失败说明方案不对）

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
using namespace std;
int a[70],vis[70];
int num,len,n;
int cmp(int a,int b)
{
    return a>b;
}
bool dfs(int l,int cnt,int x) {
    if (cnt==num)
        return 1;
    if (l==0)
    {
        int k=0;
        while(vis[k])
        {
            k++;
        }
        vis[k]=1;
        if (dfs(a[k],cnt,k+1))
            return 1;
        vis[k]=0;
        return 0;
    }
    if (l==len)
        return dfs(0,cnt+1,0);
    for (int i=x;i<n;)
    {
        if(!vis[i]&&l+a[i]<=len)
        {
            vis[i]=1;
            if(dfs(l+a[i],cnt,i+1))
            return 1;
            vis[i]=0;
            int t=a[i];
            while(i<n&&a[i]==t)i++;
        }
        else i++;
    }
    return 0;
}
int main()
{
    while (scanf("%d",&n),n)
        {
        int sum=0,flag=1;
        for (int i=0;i<n;i++)
           {
                scanf("%d",&a[i]);
                sum+=a[i];
           }
        sort(a,a+n,cmp);
        int t=sum/2;
        for (len=a[0];len<=t&&flag;len++)
        {
            if(sum%len==0)
            {
                memset(vis,0,sizeof(vis));
                num=sum/len;
                if (dfs(0,0,0))
                {
                    flag=0;
                    break;
                }
            }
        }
        printf("%d\n",flag?sum:len);
    }
    return 0;
}