LeetCode 543. 二叉树的直径 | C++语言版

最新推荐文章于 2024-07-09 11:50:42 发布

原创最新推荐文章于 2024-07-09 11:50:42 发布 · 462 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #c++ #算法

文章介绍了如何用C++解决LeetCode上的543题，即计算二叉树的直径。提供了两种解题思路，一是使用递归方法计算每个节点的深度并更新最大直径，二是优化方法以减少遍历的节点数。代码实现包括了递归版本和优化版本，并给出了参考链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LeetCode 543. 二叉树的直径 | C++语言版

- LeetCode 543. 二叉树的直径

LeetCode 543. 二叉树的直径

题目描述

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

在这里插入图片描述

解题思路

思路一：使用递归

在这里插入图片描述

代码实现

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 Max;

    int depth(TreeNode* rt){
        if(rt==NULL){
            //递归结束的出口
            return 0;
        }

        //当前节点rt左子树的深度
        int L=depth(rt->left);
        //当前节点rt右子树的深度
        int R=depth(rt->right);
        
        //每次递归都更新下局部变量
        if(L+R>Max) Max=L+R;
        
        //返回当前节点rt的深度（取左子树/右子树的最大深度+1）
        return max(L,R)+1;
    }

    
    int diameterOfBinaryTree(TreeNode* root) {
        //二叉树的直径长度是任意两个结点路径长度中的最大值（注意：两结点之间的路径长度是以它们之间边的数目表示）
        //两个叶子结点之间路径=根节点左右儿子的深度之和
        Max=0;
        depth(root);
        return Max;
    }
};