leetcode简单题 -- 22.二叉树的直径

最新推荐文章于 2022-11-15 09:55:27 发布

天空12138

最新推荐文章于 2022-11-15 09:55:27 发布

阅读量94

点赞数

分类专栏：算法文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/qq_40646399/article/details/124511809

版权

算法专栏收录该内容

59 篇文章 0 订阅

订阅专栏

本文介绍了如何使用深度优先搜索(DFS)算法解决二叉树的最大直径问题。通过递归遍历二叉树的每个节点，计算以每个节点为起点的最长路径，最终得到整棵树的直径。在JavaScript中，通过定义一个辅助函数dfs计算节点的最大深度，并在遍历过程中更新全局最大直径。最后返回的结果是去掉路径长度1的直径。

摘要由CSDN通过智能技术生成

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

示例1：给定二叉树
          1
         / \
        2   3
       / \     
      4   5  
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

解题思路：深度优先遍历

一条路径长度 = 路径上节点数量 - 1
任一条路径等于由某个节点作为起点，从其左孩子和有孩子向下遍历的路径拼接而成
假设已知节点node, 其左孩子向下遍历的最大节点数为left, 其右孩子向下遍历的节点数为right, 则以node为起点的路径经过节点数最大为left + right + 1
定义一个函数dfs(node), 用来表述以node为顶点的树最大深度，递归计算其左右子树的深度L和R, 该函数可以返回max(L, R) + 1

代码实现（js）

/**
 * 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) {
    // 1.定义遍历
    let res = 1
    // 2.深度遍历
    const dfs = (node) => {
        // 2.1 首先判断node是否为空
        if(node === null){
            return null
        }
        // 2.2 递归遍历node的左右子树
        let left = dfs(node.left)
        let right = dfs(node.right)
        // 2.3.计算以该节点为起点的路径经过节点数的最大值
        res = Math.max(res, left + right + 1)
        // 2.4 返回的是高度
        return Math.max(left, right) + 1
    }
    dfs(root)
    return res - 1
};