二叉樹平衡檢查（牛客網）

題目描述

實現一個函數，檢查二叉樹是否平衡，平衡的定義如下，對於樹中的任意一個結點，其兩顆子樹的高度差不超過1。

給定指向樹根結點的指針TreeNode* root，請返回一個bool，代表這棵樹是否平衡。

---

思路：

1、先求出左右兩顆子樹的高度，再判斷是否平衡

2、遞歸大法

貼代碼：

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) {
        // write code here
        if(root == null){
            return true;
        }
        int leftHeight = getTreeHeight(root.left);
        int rightHeight = getTreeHeight(root.right);
        if(Math.abs(leftHeight - rightHeight) > 1){
            return false;
        }
        return isBalance(root.left) && isBalance(root.right);
    }
    
    public static int getTreeHeight(TreeNode root){
        if(root == null){
            return 0;
        }
        return Math.max(getTreeHeight(root.left), getTreeHeight(root.right)) + 1;
    }
}