LeetCode.572.另一棵树的子树

题目描述:

给你两棵二叉树 rootsubRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false

二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树

输入输出实例:

思路:这道题目不绕,我们只需要使用深度优先搜索,遍历root树,然后找subroot是否与root本身、或与其左子树或者是右子树上的一部分相同,我们使用递归方法。构建一个函数isSubtree2,将root和subroot两个树作为参数,如果root为空我们返回False,如果root与subroot树相同我们就返回True,这两种情况讨论完我们就递归root的左子树或者是右子树看它是否与subroot树相同。所以我们还需要一个用来判断两个树是否相同的函数isSameTree,参数也是两个数,如果两个树都为空我们返回True,然后看是否有一个为空树,如果是我们返回False,讨论完这两种情况,我们需要return (s.val == t.val) and isSameTree(s.left,t.left) and isSameTree(s.right,t.right),即如果当前s节点值与t节点值相同,s左孩子与t左孩子值相同并且s右孩子与t右孩子相同,我们就返回True。根据上述思路,有以下代码:

python 复制代码
# 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
def isSameTree(s,t):
    #都是空树,直接返回True
    if not s and not t:
        return True
    #如果其中一个是空,返回False
    if not s or not t:
        return False
    #该节点和左子树右子树都相同返回True
    return (s.val == t.val) and isSameTree(s.left,t.left) and isSameTree(s.right,t.right)

def isSubtree2(r,su):
    if not r:
            return False
    if isSameTree(r,su):
            return True
    return isSubtree2(r.left,su) or isSubtree2(r.right,su)

class Solution:
    def isSubtree(self, root: Optional[TreeNode], subRoot: Optional[TreeNode]) -> bool:
        #思路:遍历root树,找root本身或者子树中是否有与subRoot相同的
        return isSubtree2(root,subRoot)
相关推荐
Bigcrab__2 分钟前
Python3网络爬虫开发实战(15)Scrapy 框架的使用(第一版)
爬虫·python·scrapy
weixin_4866811429 分钟前
C++系列-STL容器中统计算法count, count_if
开发语言·c++·算法
一道秘制的小菜36 分钟前
C++第七节课 运算符重载
服务器·开发语言·c++·学习·算法
易辰君1 小时前
Python编程 - 协程
开发语言·python
宇宙第一小趴菜1 小时前
探索网络世界:TCP/IP协议、Python Socket编程及日常生活比喻
网络·python·tcp/ip
小黑031 小时前
Phoenix使用
开发语言·python
南斯拉夫的铁托1 小时前
(PySpark)RDD实验实战——求商品销量排行
python·spark·pyspark
kay_5451 小时前
YOLOv8改进 | 模块缝合 | C2f 融合SCConv提升检测性能【CVPR2023】
人工智能·python·深度学习·yolo·目标检测·面试·yolov8改进
kngines1 小时前
【PLW004】基于Python网络爬虫与推荐算法的新闻推荐平台v1.0(Python+Django+NLP+Vue+MySQL前后端分离)
爬虫·python·nlp·推荐算法
激动的兔子2 小时前
Python_两个jpg图片文件名称互换
python