剑指offer——平衡二叉树

题目描述

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

 

解题思路：

如果子树是平衡二叉树，则返回子树的高度；如果发现子树不是平衡二叉树，则直接停止遍历，

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return IsBalanced(pRoot)!=-1;
    }
    //返回-1表示以某个节点为根的子树不是平衡二叉树，那么这棵树就不是平衡二叉树
    int IsBalanced(TreeNode* pRoot){
        if(pRoot==nullptr) return 0;
        int left,right;
        left=IsBalanced(pRoot->left);
        if(left==-1) return -1;
        right=IsBalanced(pRoot->right);
        if(right==-1) return -1;
        if(abs(left-right)>1) return -1;
        else return left>right?left+1:right+1;
    }
};