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

李莲花*

于 2023-04-01 18:53:45 发布

阅读量169

点赞数

文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/weixin_45992021/article/details/129898821

版权

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