牛客题霸 [ 树的直径] C++题解/答案

题目描述

给定一棵树，求出这棵树的直径，即两个节点距离的最大值。

题解：

不知道大家听没听过一个结论：
树的直径可以通过两边dfs找到
步骤：
1.从任意一点进行dfs，然后找到一个最长路径，记录最远点u
2.然后从u再进行dfs，找最长路径，记录一点v。
（u，v）就是树的直径
证明：

我们可以看出图中，树的直径是（4->2->5），长度为9.
我们一开始选定一个点dfs
如果是直径外一点，比如w1，从w1进行dfs找到的就是点4，路径就1->2->4，这个路径一定会与树的直径相交，而找到的4是直径的一端，那从4再进行dfs就是树的直径的另一端5，这样两遍dfs你就找到了树的直径
如果是直径内一点，比如w2，从w2开始dfs找到的最远点4，这个路径会被包含在树的直径里，那找到的点也就是树直径的一端，再dfs就可以找到另一端。
综上用两遍dfs就可以找到树的直径

其实求直径也就是求深度，所以我们dfs直接求树的深度，并记录最大深度，以及对应的节点，然后再带入到第二遍dfs即可

代码：

/*** struct Interval {*  int start;* int end;* };*/class Solution {public:/*** 树的直径* @param n int整型 树的节点个数* @param Tree_edge Interval类vector 树的边* @param Edge_value int整型vector 边的权值* @return int整型*/int far,Max;vector<vector<pair<int,int> >>G;void dfs(int u,int deep,int fa){if(deep>Max){far=u;Max=deep;}for(auto &it:G[u]){int v=it.first;int w=it.second;if(v==fa)continue;dfs(v,deep+w,u);}}int solve(int n, vector<Interval>& Tree_edge, vector<int>& Edge_value) {// write code hereG.clear(); G.resize(n + 1);for (int i = 0; i < n - 1; ++i) {Interval e = Tree_edge[i];int w = Edge_value[i];int u=e.start;int v=e.end;G[u].push_back(pair<int,int>(v, w));G[v].push_back(pair<int,int>(u, w));}far = 1; Max = 0;dfs(1, 0, 0);dfs(far, 0, far);return Max;}
};

牛客题霸 [ 树的直径] C++题解/答案相关推荐

牛客题霸 [二叉树的镜像]C++题解/答案
牛客题霸 [二叉树的镜像]C++题解/答案题目描述操作给定的二叉树,将其变换为源二叉树的镜像. 题解: 根据题也不难看出,二叉树的镜像定义其实就是左右树翻转所以我们一直递归左子树,到头后递归右子 ...
牛客题霸 [矩阵元素查找] C++题解/答案
牛客题霸 [矩阵元素查找] C++题解/答案题目描述已知int一个有序矩阵mat,同时给定矩阵的大小n和m以及需要查找的元素x,且矩阵的行和列都是从小到大有序的.设计查找算法返回所查找元素的二元数 ...
牛客题霸 [栈和排序] C++题解/答案
牛客题霸 [栈和排序] C++题解/答案题目描述给你一个1->n的排列和一个栈,入栈顺序给定你要在不打乱入栈顺序的情况下,对数组进行从大到小排序当无法完全排序时,请输出字典序最大的出栈序 ...
牛客题霸 [字符串的排列] C++题解/答案
牛客题霸 [字符串的排列] C++题解/答案题解: stl真好用,emmm... 代码: class Solution {public:vector<string> Permutatio ...
牛客题霸 [ 验证IP地址] C++题解/答案
牛客题霸 [ 验证IP地址] C++题解/答案题目描述编写一个函数来验证输入的字符串是否是有效的 IPv4 或 IPv6 地址 IPv4 地址由十进制数和点来表示,每个地址包含4个十进制数,其范围 ...
牛客题霸 [表达式求值] C++题解/答案
牛客题霸 [表达式求值] C++题解/答案题目描述请写一个整数计算器,支持加减乘三种运算和括号. 题解: 没有除法emmm 我们从头开始依次判断每个字符如果是左括号,我们就找右括号,并截取括号内 ...
牛客题霸 [判断回文] C++题解/答案
牛客题霸 [判断回文] C++题解/答案题目描述给定一个字符串,请编写一个函数判断该字符串是否回文.如果回文请返回true,否则返回false. 题解: 左右两端同时向中间缩代码: class ...
牛客题霸 [分糖果问题] C++题解/答案
牛客题霸 [分糖果问题] C++题解/答案题目描述一群孩子做游戏,现在请你根据游戏得分来发糖果,要求如下: 每个孩子不管得分多少,起码分到一个糖果. 任意两个相邻的孩子之间,得分较多的孩子必须拿多 ...
牛客题霸 [合并有序链表] C++题解/答案
牛客题霸 [合并有序链表] C++题解/答案题目描述将两个有序的链表合并为一个新链表,要求新的链表是通过拼接两个链表的节点来生成的. 题解: 首先判断l1和l2是否为空然后依次比较l1和l2的值 ...

牛客题霸 [ 树的直径] C++题解/答案

题目描述

题解：

代码：

牛客题霸 [ 树的直径] C++题解/答案相关推荐

最新文章

热门文章