其他分享
首页 > 其他分享> > 二叉树的几道相似简单递归题

二叉树的几道相似简单递归题

作者:互联网

二叉树中递归的思想,在这本Leetbook中讲的很细了,这里不展开。下面是几道例题:

100. 相同的树

class Solution:
    def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
        if (not p) and (not q):
            return True
        elif (not p) or (not q):
            return False
        elif p.val != q.val:
            return False
        else:
            return self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right)

判断两个二叉树是否相同,思路如下:两个二叉树是否同时为空?是否一个为空而另一个不为空?他们的根节点值是否相同?最后是递归的判断,他们的左右子树是否都相同?

101. 对称二叉树

class Solution:
    def isSymmetric(self, root: TreeNode) -> bool:
        return self.check(root, root)

    def check(self, p: TreeNode, q: TreeNode) -> bool:
        if (not p) and (not q):
            return True
        elif (not p) or (not q):
            return False
        elif p.val != q.val:
            return False
        else:
            return self.check(p.left, q.right) and self.check(p.right, q.left)

判断一个二叉树是否为对称二叉树,可以转化为判断这个树和自己的镜像树是否相同。关键在于用 self.check(root, root) 构造出两个树,然后以相反方向进行递归和判断。

104. 二叉树的最大深度

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

自底向上的递归,在Leetbook中有详细介绍。

112. 路径总和

class Solution:
    def hasPathSum(self, root: TreeNode, targetSum: int) -> bool:
        if not root: # 非节点
            return False
        if not root.left and not root.right:  # 叶子节点
            return targetSum == root.val
        return self.hasPathSum(root.left, targetSum - root.val) or self.hasPathSum(root.right, targetSum - root.val)

思路是如果子树的和等于 targetSum 减去当前节点的值,则存在路径。所以终止条件为非节点则返回 False,为叶子节点则返回 targetSum == 当前节点的值,函数调用是子节点和减去当前节点值后的 targetSum 。

标签:right,return,递归,self,几道,二叉树,root,left
来源: https://blog.csdn.net/cnhwl/article/details/120937750