给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true 。
示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false 。
tips : 自顶向上法, 时间复杂度O(n), 空间复杂度O(n)(递归栈)
class Solution {
public:
bool isBalancedRecursive(TreeNode* root, int & height) {
int left_height = 0;
int right_height = 0;
if(root == NULL) {
return true;
}
if(isBalancedRecursive(root->left, left_height) && isBalancedRecursive(root->right, right_height) && abs(left_height - right_height) <= 1 ) {
height = max(left_height, right_height)+1;
return true;
} else {
height = max(left_height, right_height)+1;
return false;
}
}
bool isBalanced(TreeNode* root) {
int height = 0;
return isBalancedRecursive(root, height);
}
};