给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:true
示例 2:
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
示例 3:
输入:root = []
输出:true
1.自己的解法:判断每一个节点是否符合平衡二叉树的定义。计算出当前节点的左右子树的最大深度,计算差值是否超过1,如果超过1,就说明不是平衡二叉树,返回false;如果未超过1,则利用递归,继续向下判断左右子树是否为平衡二叉树。这是一种自顶向下的递归。这种方法在计算每个节点的左右子树高度时,会产生很多重复计算过程,导致时间复杂度增加。最好还是要有自底向上的思想。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean isBalanced(TreeNode root) {
if(root == null){//如果当前节点为空,则返回true
return true;
}
if(Math.abs(maxLength(root.left) - maxLength(root.right)) > 1){//如果当前节点的左右子树的深度差距大于1,返回false
return false;
}
return isBalanced(root.left) && isBalanced(root.right);//当前节点符合平衡二叉树定义,继续向下判断其左右子节点是否符合定义
}
//计算当前节点的最大深度
public int maxLength(TreeNode node){
if(node == null){
return 0;
}
return Math.max(maxLength(node.left),maxLength(node.right)) + 1;
}
}
2.答案解法:提供了一种自底向上的递归。这种方法不会重复计算节点高度,而是从底部开始计算高度,然后判断当前节点是是否符合定义,如果符合定义继续向上判断。
class Solution {
public boolean isBalanced(TreeNode root) {
return height(root) >= 0;//根节点的高度大于等于0,即不是-1的话,就返回true
}
//计算当前节点的最大深度
public int height(TreeNode node){
if(node == null){//当前节点为空,高度返回0
return 0;
}
int leftHeight = height(node.left);//计算左子树的高度
if(leftHeight == -1){//说明左子树不是平衡二叉树,直接返回-1
return -1;
}
int rightHeight = height(node.right);//计算右子树的高度
if(rightHeight == -1 || Math.abs(leftHeight - rightHeight) > 1){//右子树不是二叉树,或者当前节点不符合定义
return -1;
}else{
return Math.max(leftHeight,rightHeight) + 1;//符合定义的情况下计算当前节点的高度
}
}
}
题源:力扣