572. 另一个树的子树

题目描述

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

我的解法——递归求解（AC）

• 注意代码里注释处

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

class Solution(object):
    def myisSubtree(self, s, t):
        if(not s and not t):
            return True
        elif(not s or not t):
            return False
        else:
            flag = False
            if(s.val == t.val):
                flag = self.myisSubtree(s.left, t.left) and self.myisSubtree(s.right, t.right)
            if(self.head == t and not flag):  # 注意这里，当t为完整的树时可用
                flag = self.myisSubtree(s.left, t) or self.myisSubtree(s.right, t)
            return flag
        
    def isSubtree(self, s, t):
        """
        :type s: TreeNode
        :type t: TreeNode
        :rtype: bool
        """
        self.head = t
        return self.myisSubtree(s, t)