572. 另一个树的子树

题干

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

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

给定两个非空二叉树 s 和 t，检验 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.val 则s与t一定不相同，继而判断t是否是s.left或者s.right的子树

若s.val == t.val 则s与t可能相同（只是可能），所以此时需要先判断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博客,简书哦，里面有更多的刷题笔记以及新奇思路，等你来踩！