平衡二叉树

题目描述

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

编程思路

1.重复遍历结点多次

在遍历树的每个结点的时候，调用之前求二叉树深度的函数得到它的左右子树的深度，如果每个结点的左右子树深度的差都不超过1，那么他就是一棵平衡二叉树。

2.每个结点只遍历一次

用后序遍历的方式遍历二叉树的每一个结点，在遍历到一个结点之前就已经遍历了它的左右子树，只要在遍历每个结点的时候记录它的深度（某一结点的深度等于它到叶结点的路径的长度），就可以一边遍历一边判断每个结点是不是平衡的。

程序代码（Java语言）

1.

public class Solution {
    
   public boolean IsBalanced_Solution(TreeNode root) {
        if(root == null) 
            return true;
        int left = TreeDepth(root.left);
        int right = TreeDepth(root.right);
        int dif = left - right;
        if(dif > 1 || dif < -1)
            return false;
        
        return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
    }
    public int TreeDepth(TreeNode root) {
        if(root == null)
            return 0;
        int left = TreeDepth(root.left);
        int right = TreeDepth(root.right);
        
        return (left > right) ? (left + 1) : (right + 1);
    }
    
}

2.

public boolean IsBalanced_Solution(TreeNode root) {
        int depth = 0;
        return IsBalanced(root,depth);
    }
    public boolean IsBalanced(TreeNode root,int depth) {
        if(root == null) {
            depth = 0;
            return true;
        }
        int left = 0,right = 0;
        if(IsBalanced(root.left,left) && IsBalanced(root.right,right)) {
            int dif = left - right;
			if(dif <= 1 && dif >= -1) {
                depth = 1 + (left > right ? left:right);
                return true;
            }
        }
        return false;
    }