【18】平衡二叉树

【题目】

实现一个函数，检查二叉树是否平衡，平衡的定义如下，对于树中的任意一个结点，其两颗子树的高度差不超过1。

给定指向树根结点的指针TreeNode* root，请返回一个bool，代表这棵树是否平衡。

【代码】

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/
public class Balance {
    public boolean isBalance(TreeNode root) {
        if(root == null){
            return true;
        }
       int left = getDepth(root.left);
       int right = getDepth(root.right);
       int dif = left - right;
       if(dif < -1 || dif > 1){
           return false;
       }
       return  isBalance(root.left) && isBalance(root.right);
    }
    
    //获取任意结点的深度
    public int getDepth(TreeNode node){
        if (node == null){
            return 0;
        }
        
        int left = getDepth(node.left);
        int right = getDepth(node.right);
        return left > right ? left+1 : right+1;
    }
}