6130 树的最长路

思路:树的最长路问题可以通过两次 DFS 求解,具体思路如下:

1.第一次 DFS 求树的直径
以任意一个点为起点进行深度优先遍历(DFS),找到与该点距离最远的点 u 。
以 u 为起点进行 DFS ,找到与 u 距离最远的点 v 。
则从 u 到 v 的路径即为树的直径。

2.第二次 DFS 求每个结点的最远距离
从树的中心节点(即直径的中间节点)出发,分别给两侧 DFS ,对于经过的每个结点,记录其到直径长度的最大值。

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+50;
int n,ans[maxn],dp[maxn][3],fa[maxn],son[maxn];
vector<int> G[maxn];
void dfs1(int x)
{
    dp[x][0]=dp[x][1]=dp[x][2]=-1e9;//初始化为负无穷 
    dp[x][0]=0;//直接更新就好 
    for (int i=0;i<G[x].size();i++)
    {
        int y=G[x][i];
        if (y==fa[x]) continue;
        fa[y]=x;
        dfs1(y);//处理儿子结点 
        int v=dp[y][0]+1;//v即为根到y的距离加1 
        if (v>dp[x][0])
        {
            dp[x][2]=dp[x][1];
            dp[x][1]=dp[x][0];
            dp[x][0]=v;
            son[x]=y;//记录最长链的末端 
        }
        else if (v>dp[x][1])
        {
            dp[x][2]=dp[x][1];
            dp[x][1]=v;
        }
        else if (v>dp[x][2]) dp[x][2]=v;
    }
}
void dfs2(int x,int len)//len是x到它父亲的距离 
{
    ans[x]=max(len,dp[x][0]);//更新答案 
    for (int i=0;i<G[x].size();i++)
    {
        int y=G[x][i];
        if (y==fa[x]) continue;
        dfs2(y,max(len+1,(y==son[x]?dp[x][1]:dp[x][0])+1));//注意如果y是最长链末端的儿子,那么距离需要用次长链 
    }
}
int main()
{
    scanf("%d",&n);
    for (int i=2;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        G[i].push_back(x);
        G[x].push_back(i);
    }
    dfs1(1);
    dfs2(1,0);
    for (int i=1;i<=n;i++) printf("%d ",ans[i]);
    return 0;
}
相关推荐
晨非辰6 分钟前
【数据结构入坑指南】--《层序分明:堆的实现、排序与TOP-K问题一站式攻克(源码实战)》
c语言·开发语言·数据结构·算法·面试
hansang_IR16 分钟前
【题解】P2217 [HAOI2007] 分割矩阵 [记忆化搜索]
c++·数学·算法·记忆化搜索·深搜
Voyager_42 小时前
算法学习记录03——二叉树学习笔记:从两道题看透后序位置的关键作用
笔记·学习·算法
我搞slam7 小时前
快乐数--leetcode
算法·leetcode·哈希算法
WWZZ20258 小时前
快速上手大模型:机器学习3(多元线性回归及梯度、向量化、正规方程)
人工智能·算法·机器学习·机器人·slam·具身感知
东方佑9 小时前
从字符串中提取重复子串的Python算法解析
windows·python·算法
西阳未落9 小时前
LeetCode——二分(进阶)
算法·leetcode·职场和发展
通信小呆呆9 小时前
以矩阵视角统一理解:外积、Kronecker 积与 Khatri–Rao 积(含MATLAB可视化)
线性代数·算法·matlab·矩阵·信号处理
CoderCodingNo10 小时前
【GESP】C++四级真题 luogu-B4068 [GESP202412 四级] Recamán
开发语言·c++·算法
一个不知名程序员www10 小时前
算法学习入门---双指针(C++)
c++·算法