平衡二叉树定义:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。
思路:根据定义,我们只需要后序遍历此树,从树的叶子节点开始计算高度,只要有一个子树不满足定义就返回-1,如果满足继续向上计算高度。
class Solution { public: bool IsBalanced_Solution(TreeNode* pRoot) { return deepOfTree(pRoot) != -1; } int deepOfTree(TreeNode* pRoot) { if(pRoot == NULL) return 0; int a = deepOfTree(pRoot->left); int b = deepOfTree(pRoot->right); if(a == -1 || b == -1 || abs(a-b) > 1) return -1; //根据定义,左子树不是AVLTree或右子树不是AVLTree或自己本身就算不是AVLTree else return (a>b?a:b) + 1; } }