6130 树的最长路

news/2024/7/15 18:37:40 标签: 深度优先, 图论, 算法

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

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

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

#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;
}

http://www.niftyadmin.cn/n/5294046.html

相关文章

electron-dl用于在Electron中下载多个文件

electron-dl用于在Electron中下载多个文件 const { app, BrowserWindow, ipcMain } require(electron); const { download } require(electron-dl); const path require(path);async function createWindow() {const mainWindow new BrowserWindow();mainWindow.loadURL(h…

[足式机器人]Part4 南科大高等机器人控制课 CH12 Robotic Motion Control

本文仅供学习使用 本文参考&#xff1a; B站&#xff1a;CLEAR_LAB 笔者带更新-运动学 课程主讲教师&#xff1a; Prof. Wei Zhang 课程链接 &#xff1a; https://www.wzhanglab.site/teaching/mee-5114-advanced-control-for-robotics/ 南科大高等机器人控制课 Ch12 Robotic …

vue3 组件之间传值

vue3 组件之间传值 非常好&#xff0c;为啥突然开这样一篇博文&#xff0c;首先是因为 vue3 是未来发展的趋势。其次&#xff0c;vue 官方已经确认&#xff0c;将于2023年最后一天停止对 vue2 项目的维护&#xff0c;这个是官方发出的通知&#xff0c;并且呢&#xff0c;尤雨溪…

【MCAL】TC397+EB-tresos之MCU配置实战 - 芯片时钟

本篇文章介绍了在TC397平台使用EB-treso对MCU驱动模块进行配置的实战过程&#xff0c;主要介绍了后续基本每个外设模块都要涉及的芯片时钟部分&#xff0c;帮助读者了解TC397芯片的时钟树结构&#xff0c;在后续计算配置不同外设模块诸如通信速率&#xff0c;定时器周期等&…

Matplotlib_布局格式定方圆

文章目录 一、子图1.使用 plt.subplots 绘制均匀状态下的子图2.使用 GridSpec 绘制非均匀子图 二、子图上的方法 一、子图 1.使用 plt.subplots 绘制均匀状态下的子图 返回元素分别是画布和子图构成的列表&#xff0c;第一个数字为行&#xff0c;第二个为列 figsize 参数可以…

Java连接Mysql报错:javax.net.ssl.SSLException: Received fatal alert: internal_error

大致报错日志如下&#xff1a; The last packet successfully received from the server was 11 milliseconds ago. The last packet sent successfully to the server was 10 milliseconds ago.at sun.reflect.GeneratedConstructorAccessor275.newInstance(Unknown Source)…

OpenStack云计算(-) 简介与部署Keystone

一.OpenStack简介 什么是云计算:云计算是一种按使用量付费的模式,这种模式提供可用的、便捷的、按需的网络访问,进入可配置的计算资源共享池(资源包括网络,服务器,存储,应用软件,服务) 云计算所包含的几个层次服务&#xff1a; SaaS ( Software as a Service ) :把在线软件作…

es修改mapping映射

在Elasticsearch中&#xff0c;一旦一个字段被创建&#xff0c;它的数据类型通常是固定的&#xff0c;不能直接修改。这是因为Elasticsearch是基于倒排索引的&#xff0c;字段的数据类型在创建索引时确定&#xff0c;并且与索引的结构相关联。 然而&#xff0c;如果确实需要更…