剑指 Offer-JZ39-平衡二叉树

题目描述

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

在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树

解题思路

理论知识

平衡二叉树

平衡树
平衡树(Balance Tree，BT) 指的是，任意节点的子树的高度差都小于等于1。

排序二叉树

二叉排序树
一棵空树，或者是具有下列性质的二叉树：

1. 若左子树不空，则左子树上所有结点的值均小于它的根结点的值
2. 若右子树不空，则右子树上所有结点的值均大于它的根结点的值
3. 左、右子树也分别为二叉排序树
4. 没有键值相等的结点。

实践技巧

绝对值

abs(number)

两个值之间的更大值

max(number1, number2)

代码思路

方法和计算树的深度差不多，唯一的区别在于，计算完左右子树的深度后，如果左右子树的深度大于1，即不平衡，则返回一个特殊值；同时，在计算左（右）子树深度后，我们立刻判断深度是否为该特殊值，如果为该特殊值，说明该节点的左（右）子树不平衡，则整颗二叉树不平衡，此时我们不必再继续计算剩余步骤，而是将特殊值继续传递给上层。

代码实现

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == NULL){
            return true;
        }
        
        int treeDepth = getDepth(pRoot);
        if(treeDepth == -1){
            return false;
        }
        else{
            return true;
        }
    }
private:
    int getDepth(TreeNode* pRoot){
        // 1.参数有效性判断
        if(pRoot == NULL){
            return 0 ;
        }
        
        // 2. 获取左右子树的深度
        int leftDepth = getDepth(pRoot->left);    // 2
        if(leftDepth == -1) return -1;    // 4
        int rightDepth = getDepth(pRoot->right);    // 2
        if(rightDepth == -1) return -1;    // 4
        
        // 3. 返回值
        if(abs(leftDepth - rightDepth) > 1){
            return -1;
        }
        else{
            return max(leftDepth, rightDepth) + 1;
        }
        
        // 4.回过头去处理递归
    }
};

运行结果

运行时间：3ms
占用内存：376k