【LeetCode 572】另一个树的子树

1. 题目描述

给定两个非空二叉树 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。

2. 第一种方法

• 首先定义一个辅助函数 isSameTree，用来判断两棵树是否相等：
  （1）若两棵树都不存在，可认为是相等的，返回 True；
  （2）若两棵树，一棵为空，另一棵不为空，自然是不相等的，返回 False；
  （3）若两棵树的根节点相等，继续递归分别判断它们的左子树，右子树是否相等；
  （4）若上述情况均不满足，说明不相等，返回 False。
• 回到主函数 isSubtree：
  （1）若 t 不存在（根据题干：s 也可以看做它自身的一棵子树。），则可返回 True。
  （2）若 s 不存在，主树都没有就更别提子树了，返回 False；
  （3）初始化一个变量 result 为 False，先判断 s 和 t 的根节点是否相等，若相等则可以调用 isSameTree 继续判断 s 和 t 的左节点是否相等，右节点是否相等；
  （4）若（3）得到的 result 为 False， 那说明不符合要求，就要递归判断 s 的左子树是否和 t 相等 或 s 的右子树是否和 t 相等；
  （5）最后返回 result 即可。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

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

    def isSameTree(self, root1, root2):
        if not root1 and not root2:
            return True
        if not root1 or not root2:
            return False
        if root1.val == root2.val:
            return self.isSameTree(root1.left, root2.left) and self.isSameTree(root1.right, root2.right)
        return False

3. 第二种方法

第一种方法运行之后发现太慢，翻了一下别人的题解，发现一种很机智的做法：
直接前序遍历获得 s 和 t 的前序遍历序列 ss 和 tt，然后判断 tt 是否存在于 ss 中即可。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:
        ss = self.preorder(s)
        tt = self.preorder(t)
        return tt in ss
        
    def preorder(self, root):
        if not root:
            return '#'    # 用'#'号作为单孩子的结尾，双'#'号为叶子节点结尾
        # 用 * 作为各节点值的开头
        return '*' + str(root.val) + self.preorder(root.left) + self.preorder(root.right)