LeetCode110.平衡二叉树

题目链接：LeetCode110

题解

平衡二叉树的定义是一个二叉树每个结点的左右两个子树的高度差绝对值不超过1（可以是1），用一个结构体来保存某个结点的信息（包括是否是平衡树，高度多少），算法流程如下：

1. 判断当前结点的左子树是否为平衡二叉树
2. 判断当前结点的右子树是否为平衡二叉树
3. 判断左子树与右子树的高度差的绝对值是否大于1

假设左子树已经不是平衡二叉树了，2，3都不用判断了，3判断的前提是1，2都满足条件

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {

    public ReturnData process(TreeNode head) {
        if(head == null)
            return new ReturnData(true,0);
        
        ReturnData leftData = process(head.left);
        if(!leftData.isB)
            return new ReturnData(false,0);
        
        ReturnData rightData = process(head.right);
        if(!rightData.isB)
            return new ReturnData(false,0);
        
        if(Math.abs(leftData.h - rightData.h) > 1)
            return new ReturnData(false,0);
        return new ReturnData(true,Math.max(leftData.h,rightData.h) + 1);//因为还要算上当前结点，所以要+1
    }
    public boolean isBalanced(TreeNode root) {
        return process(root).isB;
    }
}
class ReturnData {
    boolean isB;
    int h;
    ReturnData(boolean isB,int h) {
        this.isB = isB;
        this.h = h;
    }
}