【剑指offer】平衡二叉树

题目描述

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

思路

根据平衡二叉树定义，任意节点左右子数高度不能相差超过1。因此我们需要用到求树的高度的函数，求出左子树跟右子树的高度，然后若左右子树高度差大于1则不是平衡二叉树，否则继续递归判断左右子树是否满足平衡二叉树。

代码

class Solution {
public:
    int get_depth(TreeNode* pRoot){
        if(!pRoot){
            return 0;
        }
        int leftDepth = 1+get_depth(pRoot->left);
        int rightDepth = 1+get_depth(pRoot->right);
        return max(leftDepth,rightDepth);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(!pRoot){
            return true;
        }
        int leftDepth = 1+ get_depth(pRoot->left);
        int rightDepth = 1+ get_depth(pRoot->right);
        if(abs(leftDepth-rightDepth) > 1){
            return false;
        }
        //判断左右子树
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
};