判断二叉树是否是平衡二叉树

最新推荐文章于 2023-01-04 18:48:52 发布

Waitt_

最新推荐文章于 2023-01-04 18:48:52 发布

阅读量894

点赞数

文章标签：剑指OFFER C++

本文链接：https://blog.csdn.net/Waitt_/article/details/83245455

版权

剑指OFFER题12------按牛客网通过率排序

时间：2018.10.21.1959
作者：Waitt

题目

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

时间限制：1秒空间限制：32768K 热度指数：140302

解答

平衡二叉树

它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

PS:平衡二叉树未必是二叉排序树，比如二路堆就可以实现为平衡二叉树，且非二叉排序树
PS:只有在用平衡二叉树进行查找或者排序的时候才是二叉排序树

思路1

遍历每个结点，借助一个获取树深度的递归函数，根据该结点的左右子树高度差判断是否平衡，然后递归地对左右子树进行判断。
但是这种做法虽然很简单，但在判断上层结点的时候，会多次重复遍历下层结点，增加了不必要的开销。

class Solution {
public:
    int Dep(TreeNode* pRoot)
    {
        if (pRoot==NULL)
            return 0;
        int l=Dep(pRoot->left)+1;
        int r=Dep(pRoot->right)+1;
        return max(l,r);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot==NULL)
            return 1;
        return ((abs(Dep(pRoot->left)-Dep(pRoot->right))<=1)&&
            IsBalanced_Solution(pRoot->left)&&IsBalanced_Solution(pRoot->right));
    }
};

思路2

如果改为从下往上遍历，如果子树是平衡二叉树，则返回子树的高度；如果发现子树不是平衡二叉树，则直接停止遍历，这样至多只对每个结点访问一次。

class Solution {
public:
    int Dep(TreeNode* pRoot)
    {
        if (pRoot==NULL)
            return 0;
        int l=Dep(pRoot->left);
        if(l==-1)//此处的判断能够防止底层已经判断出的负数进入迭代计算
            return -1;
        int r=Dep(pRoot->right);
        if(r==-1)//此处的判断能够防止底层已经判断出的负数进入迭代计算
            return -1;
        if(abs(l-r)>1)//此处用来判断树的平衡
            return -1;
        return max(l+1,r+1);//+1放在此处可防止-1的错误迭代计算
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if (pRoot==NULL)
            return 1;
        return (Dep(pRoot)!=-1);
    }
};