输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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、定义depth方法返回当前子树的最大深度:Math.max(depth(root.left),depth(root.right))+1;
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);
}
int depth(TreeNode root){
if(root==null) return 0;
return Math.max(depth(root.left),depth(root.right))+1;
}
}
2、中序遍历+剪枝
class Solution {
public boolean isBalanced(TreeNode root) {
return recur(root)!=-1;
}
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)<=1?Math.max(left,right)+1:-1;
}
}
题目来源
作者:Krahets
链接:https://leetcode.cn/leetbook/read/illustration-of-algorithm/9hzffg/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。