572. 另一个树的子树

题干

572. 另一个树的子树(2020/5/7每日一题)

难度简单241收藏分享切换为英文关注反馈

给定两个非空二叉树 st,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。

示例 1:
给定的树 s:

     3
    / \
   4   5
  / \
 1   2

给定的树 t:

   4 
  / \
 1   2

返回 true,因为 t 与 s 的一个子树拥有相同的结构和节点值。

示例 2:
给定的树 s:

     3
    / \
   4   5
  / \
 1   2
    /
   0

给定的树 t:

   4
  / \
 1   2

返回 false

思路

重点:对于树的操作,首先要考虑递归!

本题查找子树可以分解为两个子操作

  1. 判断右边树t 是不是左边树s的子树
  2. 判断右边树t 是不是与左边树s相等的树

s.val != t.valst一定不相同,继而判断t是否是s.left或者s.right的子树

s.val == t.valst可能相同(只是可能),所以此时需要先判断s 是否与 t相同,若相同,则返回True,若不相同,仍然需要判断t是否是s.left或者s.right 的子树。

所以对于解决这道题,需要两个函数,

第一个是题目给定的判定子树的函数(对应操作1)

第二个是比较两棵树是否相同的函数(对应操作2)

  • python代码
class Solution:
    def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:
        if not t: return True
        if not s: return False
        if self.compare(s, t):
            return True
        return self.isSubtree(s.left, t) or self.isSubtree(s.right, t)

    def compare(self, s, t):
        if not s and not t: return True
        if not (s and t): return False
        if s.val != t.val:
            return False
        return self.compare(s.left, t.left) and self.compare(s.right, t.right)

当然,前面代码的逻辑只是为了方便阅读和理解,实际操作可以简化如下:

class Solution:
    def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:
        if not t: return True
        if not s: return False
        return self.compare(s, t) or self.isSubtree(s.left, t) or self.isSubtree(s.right, t)

    def compare(self, s, t):
        if not s and not t: return True
        if not (s and t): return False
        return s.val == t.val and self.compare(s.left, t.left) and self.compare(s.right, t.right)

如果感到有帮助的话,欢迎关注我的知乎,CSDN博客,简书哦,里面有更多的刷题笔记以及新奇思路,等你来踩!

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容