LeetCode 110. Balanced Binary Tree 题解 - Java

题目链接：https://leetcode.com/problems/balanced-binary-tree/#/description

题目要求：判断一棵树是否是平衡二叉树

    最初尝试使用 | 二叉树的最大深度-二叉树的最小深度 | <= 1的方法来判断一棵二叉树是否是平衡二叉树，但是发现不可行。当二叉树是下面这种形态时，该方法不正确。

    对于上图所示的二叉树，树的最大深度是5，最小深度是3，5-3>1，但是这棵树是一颗平衡二叉树。因此上述方法不正确。

正确方法：对于每个节点，计算其左右子树的最大深度，并根据这两个深度值，判断以该节点为根的二叉树是否是平衡二叉树，如果不是，则整个二叉树不是平衡二叉树；若是，继续对其他的节点进行判断。

上述过程递归进行，从叶节点向上计算和判断。

Java代码如下：

public class Solution {
	// 该方法实际上是从低向上（递归的）计算出了所有节点的左子树深度和右子树深度，
	// 并通过比较左右子树的深度差，判断是否满足平衡条件。如果某个子树不满足平衡条件，那么整个二叉树也不满足平衡条件
	
	// 如果depth的返回值为-1表示当前子树不满足平衡二叉树条件
	// 如果depth的返回值不是-1，返回的数组表示的是当前节点的最大深度值
	public int depth(TreeNode root) {
		if (root == null) {
			return 0;
		} else {
			// 获取左子树的深度（or 如果不满足平衡条件的话，直接返回-1）
			int leftDepth = depth(root.left);
			// 获取右子树的深度（or 如果不满足平衡条件的话，直接返回-1）
			int rightDepth = depth(root.right);
			if (leftDepth == -1 || rightDepth == -1 || Math.abs(leftDepth - rightDepth) > 1) {
				return -1;
				// 如果左子树深度>右子树深度，返回较大的左子树的深度+1作为父节点的深度值
			} else if (leftDepth - rightDepth > 0) {
				return leftDepth + 1;
			} else {
				// 反之返回右子树的深度+1作为父节点的深度值
				return rightDepth + 1;
			}
		}
	}

	public boolean isBalanced(TreeNode root) {
		if (root == null) {
			return true;
		}
		return depth(root)!= -1;
	}
}