LeetCode 543. 二叉树的直径

作者:Choi Yang

更新于:6 个月前

字数统计:346 字

阅读时长:1 分钟

阅读量:

题目描述

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

示例 :

给定二叉树

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

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

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/diameter-of-binary-tree 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

通过后序遍历，每一次返回当前根节点下的左右孩子的深度最大值，然后每一次递归求一下左右孩子路径和的最大值即可。

/**
 * 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 res = 0;
  let dfs = (root) => {
    if (!root) return 0;
    // 后序遍历 先遍历左孩子
    let left = root.left && dfs(root.left) + 1;
    let right = root.right && dfs(root.right) + 1;
    // 每次求直径，比较左右孩子路径和的最大值
    res = Math.max(left + right, res);
    // 递归时候返回左右孩子深度最长
    return Math.max(left, right);
  };
  dfs(root);
  return res;
};

学如逆水行舟，不进则退

Contributors

Choi Yang