题目描述:
给你这棵「无向树」,请你测算并返回它的「直径」:这棵树上最长简单路径的 边数。
我们用一个由所有「边」组成的数组 edges 来表示一棵无向树,其中 edges[i] = [u, v] 表示节点 u 和 v 之间的双向边。
树上的节点都已经用 {0, 1, …, edges.length} 中的数做了标记,每个节点上的标记都是独一无二的。
示例 1:
输入:edges = [[0,1],[0,2]]
输出:2
解释:
这棵树上最长的路径是 1 - 0 - 2,边数为 2。
示例 2:
输入:edges = [[0,1],[1,2],[2,3],[1,4],[4,5]]
输出:4
解释:
这棵树上最长的路径是 3 - 2 - 1 - 4 - 5,边数为 4。
提示:
0 <= edges.length < 10^4
edges[i][0] != edges[i][1]
0 <= edges[i][j] <= edges.length
edges 会形成一棵无向树
方法1:
主要思路:解题汇总链接
(1)dfs;
(2)和求二叉树中任意两个结点之间的最长距离的思路一致,只不过变成了多叉树的处理;
(3)由于是找任意的两个结点之间的最长距离,故任意选择一个结点作为根节点,然后使用深度优先进行遍历,求出各个结点的到终止点的最长的两个距离,相加即为该节点所在路径中,直径最长的长度;
(4)dfs的返回值是该节点到各个终止点的长度中的最长长度加1;
(5)为了实现类似的数的遍历,构建邻接图;
class Solution {
public:
int dfs(vector<vector<int>>& graph,vector<bool>& sign,int& max_d,int index){
sign[index]=true;//将访问过的结点标识
//当前结点的两个最长的到终止点的路径长度
int max_d1=0;
int max_d2=0;
for(int& g:graph[index]){
//访问当前结点的各个邻接点
if(!sign[g]){
int max_tmp=dfs(graph,sign,max_d,g);//各个邻接点下的路径长度<