leetcode(34): 判断平衡二叉树

最新推荐文章于 2024-07-28 21:09:15 发布

fang0jun

最新推荐文章于 2024-07-28 21:09:15 发布

阅读量153

点赞数

分类专栏： leetcode 文章标签： leetcode 二叉树算法

本文链接：https://blog.csdn.net/a13352912632/article/details/108959200

版权

leetcode 专栏收录该内容

40 篇文章 1 订阅

订阅专栏

该博客主要探讨了如何判断一个二叉树是否为平衡二叉树，分别介绍了两种方法：一种是自顶向下的递归解法，时间复杂度为O(n^2)；另一种是自底向上的递归策略，效率更高，时间复杂度为O(n)。通过这两部分内容，帮助读者理解在处理二叉树问题时如何优化算法。

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

//经典leetcode简单题。。。

题目描述

在这里插入图片描述

法一：自顶向下的递归 O(n^2)

/**
 * 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){
        if(root == NULL) return 0;
        return max(height(root->left), height(root->right)) + 1;

    }

    bool isBalanced(TreeNode* root) {
        if(root == NULL) return true;
        return abs(height(root->left) - height(root->right) <= 1)  && isBalanced(root->left) && isBalanced(root->right);
    }
};

简写

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        return !root ? true : abs(depth(root->left) - depth(root->right)) <= 1 && isBalanced(root->left) && isBalanced(root->right);
    }
    int depth(TreeNode* cur) { //计算最大深度
        return !cur ? 0 : max(depth(cur->left), depth(cur->right)) + 1;
    }
};

法二：自底向上的递归 O(n)

/**
 * 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:

    //确定函数功能：放入一个根节点，先求出其左右子树的高度，来判断该根节点是否为平衡树，是则返回高度(非负数)，否则返回-1
    //确定尾头：尾 root == null ； 头
    int rightOrNot(TreeNode *root){
        if(root== NULL) return 0;

        int leftHeight = rightOrNot(root->left);
        int rightHeight =rightOrNot(root->right);

        if(leftHeight == -1 || rightHeight == -1 || abs(leftHeight - rightHeight) > 1) return -1;
        else return max(leftHeight, rightHeight) + 1;

    }

    bool isBalanced(TreeNode* root) {
       return rightOrNot(root) >= 0;
    }
};