DFS 例题

最新推荐文章于 2024-09-12 19:02:20 发布

矩阵不需要秩

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量36

点赞数

分类专栏：搜索文章标签：深度优先算法 c++ dfs

本文链接：https://blog.csdn.net/wsywsyyy/article/details/132393714

版权

搜索专栏收录该内容

5 篇文章 0 订阅

订阅专栏

本文讨论了一个关于数的划分问题的AC代码实现，指出原始代码存在时间复杂度过高的问题。通过引入更精确的剪枝条件，如在搜索过程中考虑后续组合对总和的影响，作者提供了优化的解决方案。

摘要由CSDN通过智能技术生成

P1025 [NOIP2001 提高组] 数的划分 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

AC代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 8;
int n,k;
int q[N];
long long res;
void dfs(int x,int start,int sum)
{
    if(x>k)
    {
        if(sum==n)
        res++;
        return ;
    }
    
    for(int i=start;sum+(k-x+1)*i<=n;i++)
    {
        q[x]=i;
        dfs(x+1,i,sum+i);
    }
}

int main()
{
    cin>>n>>k;
    
    dfs(1,1,0);
    
    cout<<res;
    return 0;
}

相关解释：

这里按照我们一般的思路代码应该是这样的：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 8;
int n,k;
int q[N];
long long res;
void dfs(int x,int start,int sum)
{
    if(sum>n)return ;
    if(x>k)
    {
        if(sum==n)
        res++;
        return ;
    }
    
    for(int i=start;i<=n-1;i++)
    {
        q[x]=i;
        dfs(x+1,i,sum+i);
    }
}

int main()
{
    cin>>n>>k;
    
    dfs(1,1,0);
    
    cout<<res;
    return 0;
}

这里是一般的思路，那么这个代码是没问题的，但是他的时间复杂度太大了，会t两个点，所以我们要进行剪枝，这里if(sum>n)return ;算的上是剪枝，但是太模糊了，有些可以剪的他没有剪，不足够具体。所以我们可以思考如下例子：

n=7，k=4，这里已经搜到了第三位，1，3，_，_。这里的话第三位必定是大于或等于3的，所以后面最小的组合是1，3，3，3.此时就已经是大于7了，按照我们常规思路的代码这种情况sum=4<7，所以没有剪枝掉，所以我们要进行进一步的剪枝。

就这个例子，我们可以有如下计算公式（当搜到第三位的时候）：

sum+（k-x+1）*i>=n //4+（4-3+1）*3和 7进行比较，显然前者更大，所以这种情况直接剪枝。

这里必须乘以i，因为你要把i放到你当前得位置，看看能不能放。如果你乘以start的话，start它是没有变的，所以你要乘以i。