输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
解析:本题在力扣上也是标注简单题,但是我认为在该题中还是有一些内容需要注意的。
1、平衡二叉树,也就是说对于这颗树的每一个节点,其左孩子和右孩子的高度差都应该小于等于1才可。
2、要注意一定是每一个节点都满足1中的高度差条件,切勿仅仅满足root这个节点,就着急返回,导致下面节点没有判断是否满足高度差条件。
因此,我们写代码就要两部分组成,第一就是要对每一个节点判断高度差,所以需要高度差函数。
// 判断每一个节点的高度差函数
int Test(TreeNode root){
if(root == null){
return 0;
}
int left = Test(root.left);
int right = Test(root.right);
return Math.max(left, right) + 1;
}
第二就是要遍历树中所有节点,因此需要递归调用。
boolean isBalanced(TreeNode root){
if(root == null){
return true;
}
if(Math.abs(Test(root.left)-Test(root.right)) > 1){
// 说明从root开始左右孩子高度差已经大于1,不是平衡的了
return false;
}else{
// 说明root满足,接着判断root下的子节点看看满不满足。
boolean lef = isBalanced(root.left);
boolean righ = isBalanced(root.right);
return left && right;
}
}
这两部分都实现后,我们就可以写出整体代码了:
/**
* 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;
}
if(Math.abs(Test(root.left) - Test(root.right)) > 1){
return false;
}else{
boolean lef = isBalanced(root.left);
boolean rei = isBalanced(root.right);
return lef && rei;
}
}
int Test(TreeNode r){
if(r == null){
return 0;
}
int le = Test(r.left);
int ri = Test(r.right);
return Math.max(le,ri)+1;
}
}