【Leetcode】110.平衡二叉树

题目

思路

思路1
首先能想到的是求出根节点的子树的高度 根据两个子树的高度判断是否是平衡二叉树
这种操作的时间复杂度在最坏情况下O(n^2)

思路二
**我们在求子树的高度的时候 每次递归都会判断左右子树的高度的大小 我们返回较大的那个值 代表树的高度 **

但是我们能够发现 如果在每次求高度的时候 我们求出左右子树的高度后 如果这个时候高度差的绝对值已经大于1了 此时整棵树就一定不是平衡二叉树 就无需在一次次遍历整棵树来判断是否为平衡二叉树

这种操作在最坏情况下 我们也只是遍历了一次二叉树 在最后的根节点的时候才发现了高度差不符合 时间复杂度是O(n)

代码

思路一

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null){
            return true;
        }
        int leftHight = height(root.left);
        int rightHight = height(root.right);
        //递归求出根节点左右树的高度
        return Math.abs(leftHight - rightHight)<2 && isBalanced(root.left) && isBalanced(root.right);
        //如果左右树的高度差值小于2 且左数和右数也都是平衡的返回true 否则false
    }

    public int height(TreeNode root){
        if(root == null){
            return 0;
        }
        return Math.max(height(root.left) , height(root.right)) + 1;
    }
}

思路二

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null){
            return true;
        }
        return height(root) >= 0;
        //如果有一个子树不是平衡二叉树 此时这里就是-1 返回false
    }

    public int height(TreeNode root){
    //在求高度的同时判断是否为平衡二叉树
        if(root == null){
            return 0;
        }
        int leftH = height(root.left);
        int rightH = height(root.right);
        if(leftH >= 0 && rightH >= 0 && Math.abs(leftH - rightH) <= 1){
        //如果左右树的高度差小于等于1才返回高度
            return Math.max(leftH,rightH) + 1;
        }else{
        //否则返回-1 此时在下次运行到此时时 因为返回高度是-1 则还是执行else
        //此时就能把-1传给整棵树的根节点
            return -1;
        }
    }
}