HOT40-二叉树的直径

JXH_123

已于 2023-08-23 08:08:21 修改

阅读量202

点赞数

分类专栏： leetcode最热100题文章标签：算法 leetcode 二叉树

于 2023-07-04 23:33:49 首次发布

本文链接：https://blog.csdn.net/JXH_123/article/details/131545357

版权

leetcode最热100题专栏收录该内容

99 篇文章 0 订阅

订阅专栏

文章提供了一种使用后序遍历解决二叉树直径问题的方法。通过计算每个节点的最大深度，并结合左右子树的最大深度求和，找到最长路径。C++代码展示了如何实现这个算法，遍历过程中不断更新最大路径长度。

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

leetcode原题链接：二叉树的直径

上一篇：HOT39-对称二叉树

下一篇：HOT41-二叉树的层序遍历

题目描述

给你一棵二叉树的根节点，返回该树的直径。二叉树的直径是指树中任意两个节点之间最长路径的长度。这条路径可能经过也可能不经过根节点 root 。两节点之间路径的长度由它们之间边数表示。

示例 1：

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

示例 2：

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

提示：

树中节点数目在范围 [1, 104] 内
-100 <= Node.val <= 100

解题方法：左->右->根，利用二叉树的后序遍历的方法计算二叉树中每一个节点的最大深度，在遍历过程中，用当前root的左孩子最大深度和root的右孩子最大深度求和即为经过root节点的最大直径，遍历过程中用一个遍历保存每一个节点的最大直径即可。由于遍历方式是左->右->根，所以计算深度的升序是自下而上计算的，所以不会有重复递归调用问题。

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:
    int diameterOfBinaryTree(TreeNode* root) {
        if (!root) {
            return 0;
        }
        int max_len = 0; //记录最大路径长度
        max_nodes_num(root, max_len);
        return max_len;

    }
    int max_nodes_num(TreeNode* root, int& max_path) { //返回结果root节点的最大深度
        if (!root) {
            return 0;
        }
        int max_left_nodes_cnt = max_nodes_num(root->left, max_path); //左子树的最大深度，一个节点深度算1
        int max_right_nodes_cnt = max_nodes_num(root->right, max_path);//右子树的最大深度
        max_path = std::max(max_path, max_left_nodes_cnt + max_right_nodes_cnt);//用经过root最大路径长度的值去更新max_path
        return  std::max(max_left_nodes_cnt, max_right_nodes_cnt) + 1;//返回以root节点为根节点对应的深度
    }
};