题目链接
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
返回 true 。
示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]
返回 false 。
限制:
1 <= 树的结点个数 <= 10000
题解
注意:有时为了左子树的左子树和右子树、和右子树的左子树和右子树都满足条件,需要左子树和右子树都对当前函数进行递归调用
以下两种方法均基于以下性质推出: 此树的深度 等于 左子树的深度 与 右子树的深度 中的 最大值 +1 。
方法一:先序遍历 + 判断深度 (从顶至底)
题解链接
有了计算节点高度的函数,即可判断二叉树是否平衡。具体做法类似于二叉树的前序遍历,即对于当前遍历到的节点,首先计算左右子树的高度,如果左右子树的高度差是否不超过1,再分别递归地遍历左右子节点,并判断左子树和右子树是否平衡。 这是一个自顶向下的递归的过程。
实现代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean isBalanced(TreeNode root) {
if(root == null) return true;
return Math.abs(depth(root.left)-depth(root.right)) <=1 && isBalanced(root.left) && isBalanced(root.right);
}
public int depth(TreeNode root){
if(root==null) return 0;
return Math.max(depth(root.left),depth(root.right)) + 1;
}
}
复杂度分析
时间复杂度: O(n ^2),其中 n 是二叉树中的节点个数。最坏情况下,二叉树是满二叉树,需要遍历二树中的所有节点,时间复杂度是O(n)。
对于节点 p,如果它的高度是 d,则 height§ 最多会被调用 d 次(即遍历到它的每一个祖先节点时)。对于平均的情况,一棵树的高度 h 满足 O(h)=O(logn),因为 d≤h,所以总时间复杂度为 O(nlogn)。对于最坏的情况,二叉树形成链式结构,高度为 O(n),此时总时间复杂度为 O(n ^2)。
空间复杂度:O(n),其中 n 是二叉树中的节点个数。空间复杂度主要取决于递归调用的层数,递归调用的层数不会超过 n。
注意:判断深度是是否相差1时要用Math.abs
方法二:后序遍历 + 剪枝 (从底至顶)
思路是对二叉树做后序遍历,从底至顶返回子树深度,若判定某子树不是平衡树则 “剪枝” ,直接向上返回。
算法流程:
recur(root) 函数:
返回值:
当节点root 左 / 右子树的深度差 ≤1 :则返回当前子树的深度,即节点 root 的左 / 右子树的深度最大值 +1 ( max(left, right) + 1 );
当节点root 左 / 右子树的深度差 >2 :则返回 −1 ,代表此子树不是平衡树 。
终止条件:
当 root 为空:说明越过叶节点,因此返回高度 0 ;
当左(右)子树深度为 −1 :代表此树的 左(右)子树 不是平衡树,因此剪枝,直接返回 −1 ;
isBalanced(root) 函数:
返回值: 若 recur(root) != -1 ,则说明此树平衡,返回 true ; 否则返回 false 。
class Solution {
public boolean isBalanced(TreeNode root) {
return recur(root)==-1 ? false : true;
}
//自底向上,后序遍历
public int recur(TreeNode root){
if(root==null) return 0;
int left = recur(root.left);
if(left==-1) return -1;
int right = recur(root.right);
if(right==-1) return -1;
//自底至上,若出现不平衡则直接剪枝
return Math.abs(left-right)<2 ? Math.max(left,right)+1 : -1;
}
}