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

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

解答

  1. /**
  2. * Definition for a binary tree node.
  3. * function TreeNode(val, left, right) {
  4. * this.val = (val===undefined ? 0 : val)
  5. * this.left = (left===undefined ? null : left)
  6. * this.right = (right===undefined ? null : right)
  7. * }
  8. */
  9. /**
  10. * @param {TreeNode} root
  11. * @return {number}
  12. */
  13. const getDiameterByNode = (node, deep = 0) => {
  14. if (!node) return deep;
  15. return Math.max(getDiameterByNode(node.left, deep + 1), getDiameterByNode(node.right, deep + 1));
  16. }
  17. var diameterOfBinaryTree = function(root) {
  18. let max = 0;
  19. const traverse = node => {
  20. if (!node) return;
  21. const diameter = getDiameterByNode(node.left) + getDiameterByNode(node.right);
  22. max = Math.max(diameter, max);
  23. traverse(node.left);
  24. traverse(node.right);
  25. }
  26. traverse(root);
  27. return max;
  28. };