算法,从简单刷起~543. 二叉树的直径

104 阅读1分钟

本文首发于 语雀文档

题目

leetcode-cn.com/problems/di…

代码

/**
* 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 === null)
    return 0
  let res = 1
  function traverse(root) {
    if (root === null)
      return 0
    const left = traverse(root.left)
    const right = traverse(root.right)
    res = Math.max(res, left + right + 1)
    return Math.max(left, right) + 1
  }
  traverse(root)
  return res - 1
};

反思

好久没刷了,这次刷回这题竟然又忘记了。

刷这题的时候,脑子里总有一个想法阻碍了我的思考,那就是:我该如何比较某一节点与其上层节点(或根节点)的路径长度大小。
但其实不应该这么想,我们应该把问题分解圈定在一个节点内(也就是动态规划 ,拆分问题)并简单化,所以正解应该是:对于一个节点,它的最大路径长度是多少,那肯定是左子节点长度+右子节点长度+当前节点,如果想知道哪个节点的路径长度最大,那就定义个变量专门存比较后的大小便是了。