模拟卷Leetcode【剑指 Offer】Offer_55 - II. 平衡二叉树

最新推荐文章于 2023-03-16 19:19:27 发布

邂逅模拟卷

最新推荐文章于 2023-03-16 19:19:27 发布

阅读量117

点赞数

分类专栏： # leetcode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_34451909/article/details/121282358

版权

leetcode 专栏收录该内容

297 篇文章 2 订阅

订阅专栏

Offer_day18_55 - II. 平衡二叉树

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

返回 false 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ping-heng-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码：

from leetcode_python.utils import *


class Solution:
    def __init__(self):
        pass

    def maxDepth(self, root: TreeNode) -> int:
        if not root:return 0
        return 1+max(self.maxDepth(root.left),self.maxDepth(root.right))

    def isBalanced(self, root: TreeNode) -> bool:
        if not root:return True
        elif abs(self.maxDepth(root.left)-self.maxDepth(root.right))>1:
            return False
        else:
            return self.isBalanced(root.left) and self.isBalanced(root.right)


def test(data_test):
    s = Solution()
    return s.isBalanced(*data_test)


def test_obj(data_test):
    result = [None]
    obj = Solution(*data_test[1][0])
    for fun, data in zip(data_test[0][1::], data_test[1][1::]):
        if data:
            res = obj.__getattribute__(fun)(*data)
        else:
            res = obj.__getattribute__(fun)()
        result.append(res)
    return result


if __name__ == '__main__':
    datas = [
        [],
    ]
    for data_test in datas:
        t0 = time.time()
        print('-' * 50)
        print('input:', data_test)
        print('output:', test(data_test))
        print(f'use time:{time.time() - t0}s')