二叉树之平衡二叉树

1.本题知识点

   二叉树

2. 题目描述

   输入一棵二叉树，判断该二叉树是否是平衡二叉树。

3.平衡二叉树概念

   每一个节点的左右子树的高度差的绝对值不大于1，或者本身是空树。

4. 思路

   剪枝，其实就是从下到上遍历，只需要遍历一次。如果采用之前计算树深的方式，会有很多重复遍历，增加开销。

   ① 如果底层节点不是平衡二叉树，停止遍历，直接返回。

   ②如果底层节点是平衡二叉树，返回其高度(深度)。

   实现一，从上到下遍历的思路：

public boolean IsBalanced_Solution(TreeNode pRoot) {
		if (pRoot == null) {
			return true;
		}
		int left = TreeDepth(pRoot.left);
		int right = TreeDepth(pRoot.right);
		int diff = left - right;
		//判断高度差
		if (diff > 1 || diff < -1) {
			return false;
		}
		//由于从上到下的遍历思想，所以左右子节点也要同样的判断，这里就导致了底层的树深会被计算多次，增加额外开销。
		return IsBalanced_Solution(pRoot.right)
				&& IsBalanced_Solution(pRoot.left);
	}

	private int TreeDepth(TreeNode pRoot) {
		if (pRoot == null) {
			return 0;
		}
		int left = TreeDepth(pRoot.left);
		int right = TreeDepth(pRoot.right);
		return 1 + Math.max(left,right);
	}

   实现二，剪枝思路，从下到上遍历的思路：

public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        return getDepth(root) != -1;
    }
     
    private int getDepth(TreeNode root) {
        if (root == null) return 0;
        //左子树深度
        int left = getDepth(root.left);
        if (left == -1) return -1;//剪枝
        //右子树深度
        int right = getDepth(root.right);
        if (right == -1) return -1;//剪枝
        //左右子树的深度差的绝对值大于1，不是平衡二叉树，直接返回-1； 否则，返回树深
        return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
    }
}