字节跳动面试题(编程题)—平衡二叉树(思路＋代码)—力扣110

题目要求：

思路：求平衡二叉树，就要先求出树的左右子树的高度(创建一个方法)，然后判断是否满足平衡二叉树条件(另一个方法),但是这种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)