树的直径求解算法-CSDN博客

本文链接：https://blog.csdn.net/bmicnj/article/details/120162875

该博客介绍了如何求解树的直径问题，提供了一个利用两次BFS或DFS算法找到树中最远两点距离的方法。首先从任意节点开始找到最长路径的终点，然后从这个终点再次进行搜索，最终找到的路径长度即为树的直径。这种方法通过证明确保了找到的路径是最长的。

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

描述
给定一棵树，求出这棵树的直径，即树上最远两点的距离。
包含n个结点，n-1条边的连通图称为树。
示例1的树如下图所示。其中4到5之间的路径最长，是树的直径，距离为5+2+4=11
在这里插入图片描述

示例1

输入：
6,[[0,1],[1,5],[1,2],[2,3],[2,4]],[3,4,2,1,5]
返回值：
11

思路：

定义
简单路径：路径上的所有点至多只访问了一次。
树的直径：树的最长的简单路径。
算法
（1）两次BFS/DFS：先任选一个起点，BFS/DFS找到最长路的终点，再从终点进行第二次BFS/DFS找到最长路即为树的直径。
原理：设起点为u，第一次BFS/DFS找到的终点v一定是树的直径的一个端点。
证明：1）如果u是直径上的点，则v就是直径的终点（因为如果v不是的话，则必定存在另一个点w使得u到w的距离更长，则与BFS/DFS找到了v矛盾。
2）如果u不是直径上的点，则u到v必然与树的直径相交（反证），那么交点到v必然就是直径的后半段了，所以v一定是直径的一个端点，所以从v进行BFS/DFS得到的一定是直径长度。

code：

# class Interval:
#     def __init__(self, a=0, b=0):
#         self.start = a
#         self.end = b

#
# 树的直径
# @param n int整型 树的节点个数
# @param Tree_edge Interval类一维数组 树的边
# @param Edge_value int整型一维数组 边的权值
# @return int整型
#
class<

牛客 NC99 树的直径