LeetCode572题解-CSDN博客

本文链接：https://blog.csdn.net/andyjkt/article/details/107342380

LeetCode 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。s中2节点有左子树0，t中没有，结构不完全相同

解析：
一个树是另一个树的子树则：
要么这两个树相等，要么同为空，要么相等
要么这个树是左树的子树
要么这个树hi右树的子树

class Solution:
    def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:
        if not s:　# s为空
            if not t:　#　如果t也为空，返回true，也即空树是空树的子树
                return True
            else:　# 如果t不为空，返回false，非空树不可能是空树的子树
                return False
        # 判断s是否和t相等，或者t是s的左树的子树，或者t是s的右树的子树，注意子树调用的是本函数，而不是dfs
        return self.dfs(s, t) or self.isSubtree(s.left, t) or self.isSubtree(s.right, t)
    
    def dfs(self, a, b):
        if not a and not b:　# 两者都为空，返回true
            return True
        if a and b:　# 两者都不为空，判断两者根的值是否相等，并且判断左右子树是否也相等
            return a.val == b.val and self.dfs(a.left, b.left) and self.dfs(a.right, b.right)
        # 二者其中一个为空，那么直接返回false，因为两者结构要完全相等，见示例二
        return False