110. 平衡二叉树
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/balanced-binary-tree
著作权归领扣网络所有。
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:true
示例 2:
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
示例 3:
输入:root = []
输出:true
提示:
树中的节点数在范围 [0, 5000] 内
-104 <= Node.val <= 104
自顶向下-解法
空间复杂度 O(n^2)
空间复杂度 O(1)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int height(struct TreeNode * root) {
if ( root == NULL ) {
return 0;
}
return fmax( height(root->left), height(root->right) ) + 1;
}
bool isBalanced(struct TreeNode* root){
if ( root == NULL ) {
return true;
}
return fabs( height(root->left) - height(root->right) ) <= 1 && isBalanced(root->left) && isBalanced(root->right);
}
自底向上
空间复杂度 O(n)
空间复杂度 O(1)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int height(struct TreeNode * root) {
if ( root == NULL ) {
return 0;
}
int left = height(root->left);
int right = height(root->right);
if ( left == -1 || right == -1 || fabs(left-right) > 1 ) {
return -1;
} else {
return fmax(left, right) + 1;
}
}
bool isBalanced(struct TreeNode* root) {
return height(root) >= 0;
}