LeetCode第543题:二叉树的直径

201 阅读1分钟

题干

给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 : 给定二叉树

       1
      / \
     2   3
    / \     
   4   5    

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意:两结点之间的路径长度是以它们之间边的数目表示。

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/di… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法:深度优先搜索

这道题的根本我们为是寻找两节点路径长度的最大值。由于任意一个结构都可以被看成是由同一个根节点为起点,所以我们要计算从这一根节点出发的左右节点深度之和其实就是我们要找的。

体中告诉我们可以从根节点出发,也可以不从根节点出发,如果必须经过根节点的话那我们只需要在递归时将左右树之和相加即可,但是体重告诉我们并不需要,那么我们就可以设置一个最大值,当每次遇到比当前最大值大的元素直接记录即可。

就像下图:

543.jpg

代码实现:

 /**
  * @param {TreeNode} root
  * @return {number}
  */
 var diameterOfBinaryTree = function (root) {
     var Max = 0;
     function doFind(root) {
         if (root == null) return 0;
         let num1 = doFind(root.left);
         let num2 = doFind(root.right);
         Max = Math.max(num2 + num1, Max)
         return Math.max(num1, num2) + 1
     }
     doFind(root);
     return Max;
 };

\