ChoDocs

外观

LeetCode 543. 二叉树的直径 ​

作者:Choi Yang
更新于:17 小时前
字数统计:345 字
阅读时长:1 分钟

题目描述 ​

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

示例 :

javascript
给定二叉树

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

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

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

解题思路 ​

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

javascript
/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
function diameterOfBinaryTree(root) {
  if (!root)
    return 0
  let res = 0
  const dfs = (root) => {
    if (!root)
      return 0
    // 后序遍历 先遍历左孩子
    const left = root.left && dfs(root.left) + 1
    const right = root.right && dfs(root.right) + 1
    // 每次求直径，比较左右孩子路径和的最大值
    res = Math.max(left + right, res)
    // 递归时候返回左右孩子深度最长
    return Math.max(left, right)
  }
  dfs(root)
  return res
}
javascript
学如逆水行舟，不进则退

Contributors ​

Choi Yang
文章作者：Choi Yang
文章链接：
版权声明： 本博客所有文章除特别声明外，均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 ChoDocs！