数字转化——线性筛和树形dp

束照

于 2022-11-10 20:53:23 发布

阅读量103

点赞数

CC 4.0 BY-SA版权

分类专栏：算法基础学习文章标签：算法深度优先 c++

本文链接：https://blog.youkuaiyun.com/Angus5201314/article/details/127796232

算法基础学习专栏收录该内容

51 篇文章

订阅专栏

本文介绍了一种利用递归深度优先搜索算法，通过枚举数字并构建树状结构，来找出满足特定条件的数字组合，从而求解树的最大直径问题。代码展示了如何遍历、添加节点和计算直径的过程。

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

数字转化

思路：
将每个数字枚举一下，然后找到符合条件的数字，构成一棵树，在树中寻找直径即可

代码：

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
const int N  = 5e4+10;
int sum[N];
vector<int> v[N];
bool st[N];
int ans;

int dfs(int u){
    int d1=0,d2= 0;
    for(auto i:v[u]){
        int d = dfs(i)+1;
        if(d >= d1) d2 = d1,d1 =d;
        else if(d>d2) d2 =d;
    }
    ans = max(ans,d1+d2);
    return d1;
}

int main() {
    int n;
    cin>>n;
    for(int i = 1;i<=n;i++){
        for(int j = 2;j<=n/i;j++){
            sum[i*j] += i;
        }
    }
    for(int i = 2;i<=n;i++){//必须从2开始，因为1的点会把0（sum【i】）也加进去
        if(i>sum[i])
        {
            v[sum[i]].push_back(i);
            st[i] = true;
        }}
    for(int i = 1;i<=n;i++){
        if(!st[i])
            dfs(i);
    }

    cout<<ans;



    return 0;
}