LeetCode(top100)二叉树的直径

二叉树的直径

题目描述

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

题目分析

示例 :
给定二叉树

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

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

思路（一）

（1）写出一个求树高度的函数height，每次把节点当作参数传入，求出此节点向下的高度。
（2）求以某个树为顶点，左边树高+右边树高+1（算出的直径长度，必然有顶点）
（3）遍历递归整棵树，求以每一个节点为顶点的直径长度，每次总返回最大的那个路径
（4）最终返回最大直径长度

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var diameterOfBinaryTree = function(root) {
     if(!root) return 0;
    let ans=height(root.left)+height(root.right); //将左右子树分别传到计算深度的函数中计算得到最长左子树+最长右子树+1
    return Math.max(ans,diameterOfBinaryTree(root.left),diameterOfBinaryTree(root.right)); //遍历将每个节点都传入函数中，并与ans进行比较，返回最大值
}; 

    //计算二叉树的深度
    function height(node){ 
        if(!node) return 0;
        return 1+Math.max(height(node.left),height(node.right));
    }

思路（二）

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var diameterOfBinaryTree = function(root) {
     // 默认为1是因为默认了根节点自身的路径长度
    let ans = 1;

    function depth(rootNode) {
        if (!rootNode) {
            // 如果不存在根节点，则深度为0
            return 0;
        }
        // 递归，获取左子树的深度
        let L = depth(rootNode.left);
        // 递归，获取右子树的深度
        let R = depth(rootNode.right);

        /* 关键点1
        L+R+1的公式是如何而来？
        等同于：左子树深度(节点个数) + 右子树深度（节点个数） + 1个根节点
        便是这株二叉树从最左侧叶子节点到最右侧叶子节点的最长路径
        类似于平衡二叉树的最小值节点到最大值节点的最长路径
        之所以+1是因为需要经过根节点
         */
        // 获取该树的最长路径和现有最长路径中最大的那个
        ans = Math.max(ans, L + R + 1);
        /* 关键点2
        已知根节点的左右子树的深度，
        则，左右子树深度的最大值 + 1，
        便是以根节点为数的最大深度*/
        return Math.max(L, R) + 1;
    }

    depth(root);

    // 由于depth函数中已经默认加上数节点的自身根节点路径了，故此处需减1
    return ans - 1;
};