题目要求:
思路:求平衡二叉树,就要先求出树的左右子树的高度(创建一个方法),然后判断是否满足平衡二叉树条件(另一个方法),但是这种O(n)达到了n^2 因为在求高度的时候就可能已经出现了不平衡(遍历一遍),但是得在判断时候为平衡二叉树的时候才能找到不平衡(遍历两边),遍历了两次达到了n^2,在面试中时不会通过的,所以我们可以在计算高度的时候就判断是否满足条件,这样就达到了时间复杂度是n,只遍历一遍。
1.创建一个方法用来计算左右子树的高度,为平衡树的时候才会返回高度,否则返回-1.
ps:Math.abs计算绝对值 Math.max计算最大值
public int maxDepth(TreeNode root) {
if(root == null){
return 0;
}
int leftH = maxDepth(root.left);//接收返回值 为-1时不再判断是否为平衡二叉树,会一直返回-1.
int rightH = maxDepth(root.right);
if(leftH >=0 && rightH >=0 && Math.abs(leftH - rightH) <= 1){
return Math.max(leftH,rightH)+1;//+1返回的是根节点的高度,可以想象根节点左右子树为空时0+0+1
}else{//不是平衡二叉树
return -1;
}
2.接收maxDepth的返回值,如果为正,说明有高度,返回true
public boolean isBalanced(TreeNode root) {
if(root == null){//空树也是平衡二叉树
return true;
}
return maxDepth(root) > 0;
}
完整代码👇
class Solution {
public int maxDepth(TreeNode root) {
if(root == null){
return 0;
}
int leftH = maxDepth(root.left);
int rightH = maxDepth(root.right);
if(leftH >=0 && rightH >=0 && Math.abs(leftH - rightH) <= 1){
return Math.max(leftH,rightH)+1;
}else{
return -1;
}
}
public boolean isBalanced(TreeNode root) {
if(root == null){//空树也是平衡二叉树
return true;
}
return maxDepth(root) > 0;
}
}
运行结果:图①O(n*n) ,图②O(n)