Leetcode 543. 二叉树的直径

最新推荐文章于 2025-12-24 17:58:28 发布

原创最新推荐文章于 2025-12-24 17:58:28 发布 · 598 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #深度优先 #c++

本文探讨了一种高效的求解二叉树直径的方法，通过递归计算左右子树的高度并取最大值得到直径。高度函数被应用于diameterOfBinaryTree函数中，适用于空树和非空树的情况。

请添加图片描述

/**
 * 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 height(TreeNode* root, int & rst){
        if(root == nullptr)
        return 0;
        int ld =height(root->left, rst), rd = height(root->right, rst);
        rst = max(rst,ld + rd);
        return 1 + max(ld,rd);
    }
    int diameterOfBinaryTree(TreeNode* root) {
        if(root == nullptr)
        return 0;
        int rst = 0;
        height(root, rst);
        return rst;

    }
};