题目描述
实现一个函数,检查二叉树是否平衡。在这个问题中,平衡树的定义如下:任意一个节点,其两棵子树的高度差不超过 1。
示例1
给定二叉树 [3,9,20,null,null,15,7] 3 / \ 9 20 / \ 15 7 返回 true 。
示例2
给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false 。
做题思路
递归计算树的深度,在递归计算左右子树深度的同时判断当前节点所在树是否平衡,即判断左右子树深度差的绝对值是否大于1。
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
boolean flag=true;
public boolean isBalanced(TreeNode root) {
dfs(root);
return flag;
}
//计算树的深度
public int dfs(TreeNode root){
if(root==null) return 0;
int left=dfs(root.left); //左子树深度
int right=dfs(root.right); //右子树深度
//判断本结点所在树是否平衡
if(Math.abs(left-right)>1){
flag=false;
}
//返回本结点所在树的深度(子树的深度+根节点)
return Math.max(left,right)+1;
}
}