【leetcode】110. 平衡二叉树（python）自顶向下的递归 + 自底向上的递归

最新推荐文章于 2022-10-20 19:47:20 发布

zzamx

最新推荐文章于 2022-10-20 19:47:20 发布

阅读量373

点赞数

分类专栏： Leetcode Python 文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/xiaoyue_/article/details/124736906

版权

Leetcode 同时被 2 个专栏收录

119 篇文章 2 订阅

订阅专栏

Python

70 篇文章 0 订阅

订阅专栏

在这里插入图片描述

方法一：自顶向下的递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def dfs(node):
            if not node:
                return 0
            l = dfs(node.left)
            r = dfs(node.right)
            return max(l, r) + 1

        if not root:
            return True

        if not self.isBalanced(root.left) or not self.isBalanced(root.right):
            return False

        return True if abs(dfs(root.left) - dfs(root.right)) <= 1 else False

方法二：自底向上的递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def dfs(self, node):
            if not node:
                return 0
            l = self.dfs(node.left)
            r = self.dfs(node.right)
            if abs(l - r) > 1 or l == -1 or r == -1:
                return -1
            else:
                return max(l, r) + 1

    def isBalanced(self, root: TreeNode) -> bool:
        if not root:
            return True
        return self.dfs(root) > 0