Lstmxx的空间

543. 二叉树的直径

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

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */


function diameterOfBinaryTree(root: TreeNode | null): number {
  if (root === null) return 0
  let ans = 1
  function traverse(node: TreeNode | null) {
    if (node === null) return 0
    const l = traverse(node.left)
    const r = traverse(node.right)
    const path = l + r + 1
    ans = Math.max(path, ans)
    return Math.max(l, r) + 1
  }
  traverse(root)
  return ans - 1
};