Leetcode刷题110. 平衡二叉树

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：true
示例 2：

输入：root = [1,2,2,3,3,null,null,4,4]
输出：false
示例 3：

输入：root = []
输出：true

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/balanced-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
    public boolean isBalanced(TreeNode root) {
//		return isBalancedI(root);
		return isBalancedII(root);
	}

	//方法二：不使用全局变量，思路与方法一类似
	//时间和空间复杂度O(N)
	private boolean isBalancedII(TreeNode root) {
		if (root == null) {
			return true;
		}
		return getHeight(root) != -1;
	}

	private int getHeight(TreeNode root) {
		if (root == null) {
			return 0;
		}
		//如果左子节点不是平衡二叉树，直接返回
		int left = getHeight(root.left);
		if (left == -1) {
			return -1;
		}
		//如果右子节点不是平衡二叉树，直接返回
		int right = getHeight(root.right);
		if (right == -1) {
			return -1;
		}
		//如果左右子节点都是平衡二叉树，但他们的高度相差大于1
		if (Math.abs(left - right) > 1) {
			return -1;
		}
		return 1 + Math.max(left, right);
	}

	//方法一：定义全局变量，使用后序遍历，每次更新变量值
	//时间复杂度O(N)，空间复杂度O(N)
    boolean isBalanced = true;
	private boolean isBalancedI(TreeNode root) {
		if (root == null) {
			return true;
		}
		traverse(root);
		return isBalanced;
	}

	//返回以root为节点的二叉树的高度
	private int traverse(TreeNode root) {
		if (root == null) {
			return 0;
		}
		int left = traverse(root.left);
		int right = traverse(root.right);
		if (Math.abs(left - right) > 1) {
			isBalanced = false;
		}
		return 1 + Math.max(left, right);
	}
}