剑指 Offer 55 - II. 平衡二叉树

最新推荐文章于 2022-06-19 19:26:02 发布

diluosixu

最新推荐文章于 2022-06-19 19:26:02 发布

阅读量72

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/u012763126/article/details/115430045

版权

平衡二叉树后序遍历剪枝深度二叉树算法

关键词由CSDN通过智能技术生成

leetcode 专栏收录该内容

34 篇文章 1 订阅

订阅专栏

剑指 Offer 55 - II. 平衡二叉树
输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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
注意：本题与主站 110 题相同：https://leetcode-cn.com/problems/balanced-binary-tree/
方法：后序遍历 + 剪枝
思路是对二叉树做后序遍历，从底至顶返回子树深度，若判定某子树不是平衡树则 “剪枝” ，直接向上返回。
算法流程：
isSub(root) 函数：
（1）返回值：

当节点root 左 / 右子树的深度差 ≤1 ：则返回当前子树的深度，即节点 root 的左 / 右子树的深度最大值 +1 （ max(left, right) + 1 ）；
当节点root 左 / 右子树的深度差 >2 ：则返回 −1 ，代表此子树不是平衡树。

（2）终止条件：

当 root 为空：说明越过叶节点，因此返回高度 0 ；
当左（右）子树深度为 −1 ：代表此树的左（右）子树不是平衡树，因此剪枝，直接返回 −1 ；

isBalanced(root) 函数：

返回值：若 isSub(root) != -1 ，则说明此树平衡，返回 true ；否则返回 false 。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def isSub(root):
            if not root:
                return 0
            left = isSub(root.left)
            # 剪枝
            if left == -1:
                return -1
            right = isSub(root.right)
            # 剪枝
            if right == -1:
                return -1
            return max(left, right) + 1 if abs(left-right) < 2 else -1

        return isSub(root) != -1