第二十一题 543. 二叉树的直径

543. 二叉树的直径

题目描述

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

示例

给定二叉树

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

注意

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

思路

递归 进行 dfs

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var diameterOfBinaryTree = function(root) {
    if(!root) return 0;
    let dep = 0;
    let dfs = function(node){
        if(!node) return 0;
        //左子树出发 
        let left = dfs(node.left);
        //右子树出发
        let right = dfs(node.right);
        //更新dep 直径长度
        dep = Math.max(dep, left + right);
        //递归是返回 长的一边
        return Math.max(left, right) + 1;
    }
    dfs(root);
    return dep;
};