平衡二叉树定义:它是一棵空树或它的左右两个子树的高度差的绝对值不超过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;
}
}
京公网安备 11010502036488号