编程练习-平衡二叉树

编程练习-平衡二叉树

判断一棵树是否为平衡二叉树

代码思路：
首先先设计一个计算节点深度的函数，可以得出某个节点的左节点深度和右节点深度，来判断深度差是否超过1，如果用前序遍历，会导致重复计算，所以应该改用后序遍历，这里为了保存结果要初始化一个布尔全局变量代表是否为平衡树，在遍历过程中一旦出现不平衡的子节点，就设置为false。

import java.util.*;
public class Solution {
    
    /*前序遍历版本
     private 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;
    }
    public boolean IsBalanced_Solution(TreeNode root) {
        if(root==null)
            return true;
        int left = TreeDepth(root.left);
        int right = TreeDepth(root.right);
        int sub = left - right;
        if(sub>1||sub<-1)
            return false;
        return IsBalanced_Solution(root.left)&&IsBalanced_Solution(root.right);
    }*/
    
    /**
    *后序遍历版本
    */
    private boolean isBalanced = true;
    public boolean IsBalanced_Solution(TreeNode root) {
        getDepth(root);
        return isBalanced;
    }
    
    private int getDepth(TreeNode node){
        if(node==null)
            return 0;
        int left = getDepth(node.left);
        int right = getDepth(node.right);
        
        if(Math.abs(left-right)>1)
            isBalanced = false;
        
        return left>right?left+1:right+1;
    }
}