二叉树简单 LeetCode110. 平衡二叉树

110. 平衡二叉树

描述

给定一个二叉树，判断它是否是高度平衡的二叉树。
本题中，一棵高度平衡二叉树定义为：
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

分析

判断一个树是否是平衡二叉树，需要直到1.左右孩子是否是平衡二叉树；2.左右孩子的最大深度的差的绝对值有没有超过1.
也就是说递归需要得两种信息，一个深度，一个是boolean。
解决办法：
1.把boolean作为全局变量，每次递归都判断是否需要修改boolean。
2.用特殊的整数表示左右孩子是否是平衡二叉树，如一个负数就行。

解决方法1的代码：

class Solution {
    boolean ans = true;
    public boolean isBalanced(TreeNode root) {
        process(root);
        return ans;
    }

    public int process(TreeNode root){
        if(root == null){
            return 0;
        }
        int a = process(root.left);
        if(!ans){
            return -1;
        }
        int b = process(root.right);
        if(!ans){
            return -1;
        }
        if(Math.abs(a-b) > 1){
            ans = false;
            return -2;
        }
        return Math.max(a,b) + 1;
    }
}

解决方法2的代码：

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(process(root) == -2){
            return false;
        }
        return true;
    }

    public int process(TreeNode root){
        if(root == null){
            return 0;
        }
        int a = process(root.left);
        int b = process(root.right);
        if(a == -2 || b == -2){
            return -2;
        }
        if(Math.abs(a-b) > 1){
            return -2;
        }
        return Math.max(a,b) + 1;
    }
}