每天一道leetcode：剑指 Offer 55 - II. 平衡二叉树（适合初学者&递归）

最新推荐文章于 2024-05-20 21:56:33 发布

杜若南星

最新推荐文章于 2024-05-20 21:56:33 发布

阅读量114

点赞数 1

分类专栏：剑指Offer 文章标签： leetcode 算法职场和发展图论 c++ 数据结构迭代加深

本文链接：https://blog.csdn.net/qq_63349644/article/details/132358816

版权

剑指Offer 专栏收录该内容

39 篇文章 0 订阅

订阅专栏

今日份题目：

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

示例1

给定二叉树 `[3,9,20,null,null,15,7]`

返回 `true` 。

示例2

给定二叉树 `[1,2,2,3,3,null,null,4,4]`

返回 `false` 。

提示

`0 <= 树的结点个数 <= 10000`

题目思路

使用递归，计算树高和判断是否是平衡二叉树。递归计算树高，结束条件就是到达空节点，也就是叶子节点的后边，返回0，其他情况返回左右子树高度的最高值加一。递归判断是否为平衡二叉树，就是递归判断左右子树及其左右子子树等等是否为平衡二叉树。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution 
{
public:
    int height(TreeNode* root) //递归计算树的高度
    {
        if(root==NULL) return 0;
        else 
            return max(height(root->left),height(root->right))+1; //加一是因为还有个根节点
    }

    bool isBalanced(TreeNode* root) 
    {
        if(root==NULL) return true;
        else 
            return abs(height(root->left)-height(root->right))<=1&&isBalanced(root->left)&&isBalanced(root->right); //左右子树里都不存在非平衡树且根节点处也满足平衡二叉树的条件
    }
};