题目描述
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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,处理边界值和递归二叉树每个节点即可
算法流程:
- isBalanced(root) 函数: 判断树 root 是否平衡
- 特例处理: 若树根节点 root 为空,则直接返回 true;
- 返回值: 所有子树都需要满足平衡树性质,因此以下三者使用与逻辑& 连接;
- Math.abs(depthTree(root.left) - depthTree(root.right)) <= 1 :判断 当前子树 是否是平衡树;
- isBalanced(root.left) : 先序遍历递归,判断 当前子树的左子树 是否是平衡树;
- isBalanced(root.right) : 先序遍历递归,判断 当前子树的右子树 是否是平衡树;
- depthTree(root) 函数: 计算树 root 的深度
- 终止条件: 当 root 为空,即越过叶子节点,则返回高度 0 ;
- 返回值: 返回左 / 右子树的深度的最大值 +1 。
public boolean isBalanced(TreeNode root) {
if (root == null) {
return true;
}
if (Math.abs(depthTree(root.left) - depthTree(root.right)) == 1) {
return isBalanced(root.left) && isBalanced(root.right);
}
return false;
}
public int depthTree(TreeNode root){
if (root == null) {
return 0;
}
return Math.max(depthTree(root.left),depthTree(root.right)) + 1;
}
复杂度分析:
时间复杂度O(NlogN):最差情况下(为 “满二叉树” 时), isBalanced(root) 遍历树所有节点,判断每个节点的深度 depthTree(root) 需要遍历 各子树的所有节点。因此,总体时间复杂度 = 每层执行复杂度 × 层数复杂度 =O(N×logN) 。
空间复杂度 O(N): 最差情况下(树退化为链表时),系统递归需要使用O(N) 的栈空间。
作者:jyd
链接:https://leetcode-cn.com/problems/ping-heng-er-cha-shu-lcof/solution/mian-shi-ti-55-ii-ping-heng-er-cha-shu-cong-di-zhi/