【leetcode】572. 另一棵树的子树（python）

最新推荐文章于 2023-11-18 08:47:09 发布

zzamx

最新推荐文章于 2023-11-18 08:47:09 发布

阅读量348

点赞数

分类专栏： Leetcode Python 文章标签： leetcode 深度优先二叉树

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

版权

Leetcode 同时被 2 个专栏收录

119 篇文章 2 订阅

订阅专栏

Python

70 篇文章 0 订阅

订阅专栏

在这里插入图片描述

思路：

先整体遍历一遍root，在root上找到与subRoot的根节点相同的节点值，
再用递归（isSame函数）依次判断root的子树和subRoot的节点值和结构是否一致。

注意测试用例：
[1,1]
[1]
即，可能一棵树上有多个相同的节点值，此时只要root的左孩子1和subRoot匹配，即可返回True

# 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 isSame(self, root, subRoot):
        if not root and not subRoot:
            return True
        if not root or not subRoot or root.val != subRoot.val:
            return False
        return self.isSame(root.left, subRoot.left) and self.isSame(root.right, subRoot.right)

    def isSubtree(self, root: TreeNode, subRoot: TreeNode) -> bool:
        if not root:
            return False
        if root.val == subRoot.val:
            if self.isSame(root, subRoot):  # note测试用例：[1,1][1]  即，可能一棵树上有多个相同的节点值，此时只要root的左孩子1和subRoot匹配，即可返回True
                return True
        return self.isSubtree(root.left, subRoot) or self.isSubtree(root.right, subRoot)