HDU 3534 Tree(dfs统计树的直径的数量)

最新推荐文章于 2021-10-14 21:04:13 发布

_bubble

最新推荐文章于 2021-10-14 21:04:13 发布

阅读量593

点赞数

分类专栏： ACM 文章标签： DFS 图论

本文链接：https://blog.csdn.net/rockzh1993/article/details/49718459

版权

该博客介绍了如何利用深度优先搜索（DFS）算法来计算树的直径。内容包括通过DFS寻找以当前节点为终点的最长路径长度，并比较两条最长路径的和以找到最优解。文章详细讨论了树的直径可能存在的多种情况，如多条相同长度的最长路径、一条最长路径和多条次长路径、仅有一条路径的情况。整个解决方案的时间复杂度为O(n)。

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

求树的直径用dfs可做，每次返回以当前结点为终点、某个叶子为起点的最长路径长度，同时找最大的两条路径求和更新最优解。

具体分有多条同样长的最长路径、有一条最长路径和一条或多条次长路径、只有一条路径、叶子结点这几种情况。O（n）可做。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=100005;
int val[maxn],sz[maxn];
int best,num;
struct Edge
{
    int to,cost;
    Edge(int a,int b):to(a),cost(b) {}
};
vector<Edge> g[maxn];
int dfs(int now,int fa)
{
    int maxi1=0,maxi2=0;
    int cnt1=0,cnt2=0;
    int n1=0,n2=0;
    int rt1=0;
    for(int i=0; i<g[now].size(); ++i)
    {
        int to=g[now][i].to,cst=g[now][i].cost;
        if(to==fa)
            continue;
        int res=dfs(to,now)+cst;
        if(maxi1<res)
        {
            maxi2=maxi1;
            cnt2=cnt1;
            n2=n1;

            maxi1=res;