美文网首页
leetcode刷题之递归

leetcode刷题之递归

作者: sk邵楷 | 来源:发表于2023-01-25 20:09 被阅读0次

leetcode刷题,使用python

1, 相同的树—— 0110 递归 自顶向下的递归和自底向上的递归
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

from typing import Optional

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: Optional[TreeNode]) -> bool:
        if not root:
            return True

        def height(root):
            if not root:
                return True
            return max(height(root.left), height(root.right)) + 1

        return abs(height(root.left) - height(root.right)) <= 1 and self.isBalanced(root.left) and self.isBalanced(root.right)

# 自底向上的递归
class Solution2:
    def isBalanced(self, root: Optional[TreeNode]) -> bool:

        def height(root):
            if not root:
                return 0
            left = height(root.left)
            right = height(root.right)

            if left==-1 or right == -1 or abs(left-right)>1:
                return -1
            else:
                return max(left, right) + 1

        return True if height(root)>=0 else False

root = TreeNode(3)
a1 = TreeNode(9)
a2 = TreeNode(20)
a3 = TreeNode(15)
a4 = TreeNode(7)
root.left = a1
root.right = a2
a2.left = a3
a2.right = a4
S = Solution()
S2 = Solution2()
print(S.isBalanced(root))
print(S2.isBalanced(root))

相关文章

网友评论

      本文标题:leetcode刷题之递归

      本文链接:https://www.haomeiwen.com/subject/fhbkhdtx.html