24_110. 平衡二叉树
题目描述:
解题思路:
一、自顶向下的递归
这个类似于二叉树的前序遍历,首先判断当前节点是否符合平衡二叉树(即其左右字数的高度差是否不超过1),如果不超过1,再分别递归地遍历左右子节点,并判断左右字数是否平衡。
这也是我能想的递归了,但是当考虑其复杂度的时候,由于是自顶向下,对于同一个节点,求其高度这个过程会重复调用,越靠近二叉树底层,调用的次数越多,就导致了时间复杂度的大大提升。
二、自底向上的递归
该做法类似于后序遍历,对于当前节点,先递归的判断其左右子树是否平衡,再判断以当前节点为根的子树是否平衡。如果一颗子树是平衡的,则返回其高度(高度一定是非负数),否则返回-1。如果存在一颗子树不平衡,则整个二叉树一定不平衡。
时间复杂度从O(n2)--->O(n),空间复杂度都是O(n)
自顶向下
class Solution {
public boolean isBalanced(TreeNode root) {
if(root == null){
return true;
}
return Math.abs(height(root.left) - height(root.right)) <= 1 && isBalanced(root.left) && isBalanced(root.right);
}
public int height(TreeNode root){
if(root == null){
return 0;
}
int leftHeight = height(root.left);
int rightHeight = height(root.right);
return Math.max(leftHeight,rightHeight) + 1;
}
}
自底向上
class Solution {
public boolean isBalanced(TreeNode root) {
return height(root) >= 0 ;
}
public int height(TreeNode root){
if(root == null){
return 0;
}
int leftHeight = height(root.left);
int rightHeight = height(root.right);
if(leftHeight == -1 || rightHeight == -1 || Math.abs(leftHeight - rightHeight) > 1){
return -1;
}
return Math.max(leftHeight , rightHeight) + 1;
}
}