平衡二叉树

最新推荐文章于 2022-11-18 17:33:21 发布

overlordmax

最新推荐文章于 2022-11-18 17:33:21 发布

阅读量97

点赞数

分类专栏：剑指Offer 文章标签：二叉树数据结构

本文链接：https://blog.csdn.net/jiangdongxiaobawang/article/details/104170119

版权

剑指Offer 专栏收录该内容

84 篇文章 1 订阅

订阅专栏

平衡二叉树

1.题目描述

给定一个二叉树，判断它是否是高度平衡的二叉树。
本题中，一棵高度平衡二叉树定义为：一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
在这里插入图片描述
返回 true 。
示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]

返回 false 。

2.思路

自底向上，提前阻断。从底向上求每个节点的深度，判断左右子树的高度差是否大于1，如果大于1返回-1，后续不用再判断了。如果小于1，继续遍历其他节点。

3.代码

/**
 * 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:
    bool isBalanced(TreeNode* root) {
        return depth(root) != -1;
    }
    int depth(TreeNode* root){
        if(root == NULL){
            return 0;
        }
        int left = depth(root->left);
        if(left == -1){
            return -1;
        }
        int right = depth(root->right);
        if(right == -1){
            return -1;
        }
        return abs(left - right) < 2 ? max(left, right) + 1 : -1;
    }
};