输入一棵二叉树，判断该二叉树是否是平衡二叉树。

最新推荐文章于 2024-07-22 08:34:33 发布

pyj0011

最新推荐文章于 2024-07-22 08:34:33 发布

阅读量194

点赞数

分类专栏： c++版本剑指offer 文章标签：二叉树算法数据结构

本文链接：https://blog.csdn.net/weixin_40668866/article/details/104684727

版权

c++版本剑指offer 专栏收录该内容

66 篇文章 0 订阅

订阅专栏

题目描述
输入一棵二叉树，判断该二叉树是否是平衡二叉树。

思路1： 递归
首先写一个返回树的高度的函数
然后
判断左右子树高度差
判断左子树的左右子树高度差
判断右子树的左右子树高度差
。。。

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == NULL) return true;
        return (high(pRoot->left) - high(pRoot->right) <= 1)
            && (high(pRoot->left) - high(pRoot->right) >= -1)
            && IsBalanced_Solution(pRoot->left)
            && IsBalanced_Solution(pRoot->right);
    }
    int high(TreeNode* pRoot){
        if(pRoot == NULL) return 0;
        return 1 + (high(pRoot->left) > high(pRoot->right)? high(pRoot->left) : high(pRoot->right));
    }
};

思路2： 自下而上判断是否平衡
用递归虽然好，但是占用空间较多
在计算树的高度的函数中
加入一个对于左右子树高度差的判断
一旦出现左右子树高度差不对，也就是不平衡，就返回-1
这样避免了一部分的重复计算

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == NULL) return true;
        return high(pRoot) != -1;
    }
    int high(TreeNode* pRoot){
        if(pRoot == NULL) return 0;
        int left = high(pRoot->left);
        int right = high(pRoot->right);
        if(left - right > 1 || left - right < -1)
            return -1;
        if(left == -1 || right == -1)
            return -1;
        return 1 + (high(pRoot->left) > high(pRoot->right)? high(pRoot->left) : high(pRoot->right));
    }
};