您的位置:首页 > 财经 > 产业 > 力扣最热一百题——二叉树的直径

力扣最热一百题——二叉树的直径

2024/11/19 2:42:20 来源:https://blog.csdn.net/DDDDWJDDDD/article/details/142134512  浏览:    关键词:力扣最热一百题——二叉树的直径

目录

题目链接:543. 二叉树的直径 - 力扣(LeetCode)

题目描述

示例

提示:

解法一:深度优先搜索

实现思路:

关键点:

Java写法:

运行时间

C++写法:

运行时间

总结


题目链接:543. 二叉树的直径 - 力扣(LeetCode)

注:下述题目描述和示例均来自力扣

题目描述

给你一棵二叉树的根节点,返回该树的 直径 。

二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。

两节点之间路径的 长度 由它们之间边数表示。

示例

示例 1:

输入:root = [1,2,3,4,5]
输出:3
解释:3 ,取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。

示例 2:

输入:root = [1,2]
输出:1

提示:

  • 树中节点数目在范围 [1, 10^{4}] 内
  • -100 <= Node.val <= 100

解法一:深度优先搜索

        直径是二叉树中任意两节点间最长路径的长度。思路是通过深度优先遍历来逐个节点进行递归计算,最终找出最长路径。

实现思路:

  1. 全局变量res的作用

    • 该变量用于动态保存当前发现的最长路径。初始值设为0,但在进入根节点时被初始化为1,因为树至少包含根节点。
  2. diameterOfBinaryTree方法

    • 入口方法,接收二叉树的根节点作为输入。
    • 首先检查根节点是否为空,若为空,则二叉树的直径为0
    • 调用辅助方法depthFirst,以深度优先的方式遍历整棵树。
    • 最终返回res - 1,这是因为路径的长度是节点数减1。
  3. 深度优先遍历 depthFirst

    • 每个节点通过递归方式遍历其左子树和右子树,分别获取左子树和右子树的深度。
    • 更新全局变量res,这个过程通过比较当前节点左子树和右子树的深度和当前保存的res,取最大的那个值。res的更新逻辑是左子树深度 + 右子树深度 + 1(代表当前节点)。
    • 返回当前节点的深度,即左子树和右子树深度的最大值加1,作为给上层节点使用的深度值。

关键点:

  • 深度优先搜索 (DFS):通过递归的方式遍历每一个节点。
  • 动态维护最长路径:每次计算出左右子树的深度时,更新全局变量res,并确保保存的是最长路径。
  • 最终结果调整:二叉树的直径是节点数减去1(路径的长度是节点数减1),因此最后返回res - 1

Java写法:

/*** Definition for a binary tree node.* public class TreeNode {*     int val;*     TreeNode left;*     TreeNode right;*     TreeNode() {}*     TreeNode(int val) { this.val = val; }*     TreeNode(int val, TreeNode left, TreeNode right) {*         this.val = val;*         this.left = left;*         this.right = right;*     }* }*/
class Solution {// 定义一个全局变量用于动态保存resint res = 0;public int diameterOfBinaryTree(TreeNode root) {if(root != null){res = 1;}else{return 0;}// 深度优先遍历depthFirst(root);// 由于我们最终得到是节点的个数// 而题目要的是路径的长度,那么只需要-1即可return res - 1;}/**深度优先遍历*/public int depthFirst(TreeNode node){// 非空判断if(node == null){return 0;}// 遍历左子树int leftDepth = depthFirst(node.left);// 遍历右子树int rightDepth = depthFirst(node.right);// 动态维护res,由于我们要找的是最长两个节点的长度// 那么res的值=左子树的深度+右子树的深度+node节点(也就是1)res = Math.max(res, leftDepth + rightDepth + 1);// 返回当前树的深度给上一节点return Math.max(leftDepth,rightDepth) + 1;}
}

运行时间

C++写法:

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     TreeNode *left;*     TreeNode *right;*     TreeNode() : val(0), left(nullptr), right(nullptr) {}*     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}*     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:// 定义一个全局变量用于动态保存resint res = 0;int diameterOfBinaryTree(TreeNode* root) {if(root == NULL){return 0;}else{res = 1;}// 深度优先遍历depthFirst(root);// 由于我们最终得到是节点的个数// 而题目要的是路径的长度,那么只需要-1即可return res - 1;}/**深度优先遍历*/int depthFirst(TreeNode* node){// 非空判断,结束条件if(node == NULL){return 0;}// 遍历左子树int leftDepth = depthFirst(node->left);// 遍历右子树int rightDepth = depthFirst(node->right);// 动态维护res,由于我们要找的是最长两个节点的长度// 那么res的值=左子树的深度+右子树的深度+node节点(也就是1)res = max(res, leftDepth + rightDepth + 1);// 返回当前树的深度给上一节点return max(leftDepth,rightDepth) + 1;}
};

运行时间


总结

        其实就是深度优先搜索的一个扩展现实,多维护了一个res结果,在每次的深度优先搜索的过程中动态的维护一个res,来坚持判断当前两个最远的节点的值,这里最远的节点的值,肯定是存在于两个叶子结点之间的(其实我也在思考这句话对不对,我目前分析来说是对的,没有找到反例,不知道各位同学、大佬能不能发现这句话的问题,欢迎指正),这就是这道题目的关键点所在。

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com