输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 。
限制:
- 0 <= 树的结点个数 <= 10000
分析:
方法:后序遍历
利用后序遍历 左子树 → 右子树 → 根 的特点,记录左右子树的深度,然后比较两子树之差是否大于1即可。
时间复杂度:O(n)
空间复杂度:O(n)
/**
* 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){
//空节点返回-1
if(root == null){
return 0;
}
//获取左右子树深度
int left = dfs(root.left);
int right= dfs(root.right);
//深度相差超过1,不为平衡二叉树
if(Math.abs(left-right) > 1){
flag = false;
}
//返回最大深度+1
return Math.max(left, right)+1;
}
}
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ping-heng-er-cha-shu-lcof