题目描述
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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
解题
思路:要想知道一棵树是否是平衡的,需要满足左右子树高的差小于等于1,且左右子树分别为平衡二叉树。
方法一:自顶向下
其实就是正向思考过程,先获取左子树高,再获取右子树高,比较大小;再递归判断左、右子树是否平衡,执行步骤不变。
class Solution {
public boolean isBalanced(TreeNode root) {
if(root==null) return true;
int m = height(root.left);
int n = height(root.right);
return Math.abs(m-n)<=1 && isBalanced(root.left) && isBalanced(root.right);
}
public int height(TreeNode r){
if(r==null) return 0;
int m = height(r.left);
int n = height(r.right);
return Math.max(m, n)+1;
}
}
方法二:自底向上
方法一的方法有大量冗余,获取树高时遍历子树一遍,判断子树平衡时又遍历一遍。
所以考虑进行一次遍历就能获取树高,和树是否平衡。
从叶子节点开始,先判断树是否平衡,若平衡向上层返回树高,若不平衡则直接返回不平衡结果,中止调用。
class Solution {
public boolean isBalanced(TreeNode root) {
return helper(root)!=-1;
}
public int helper(TreeNode root){
if(root==null) return 0;
int l = helper(root.left);
if(l==-1) return -1;
int r = helper(root.right);
if(r==-1) return -1;
if(Math.abs(l-r)<=1)
return Math.max(l, r)+1;
return -1;
}
}