题目描述

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

例如:

给定的树 A:
     3
    / \
   4   5
  / \
 1   2

给定的树 B：
   4 
  /
 1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false

示例2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true

限制：
0 <= 节点个数 <= 10000

解答方法

方法一：BFS+递归

思路

https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof/solution/mian-shi-ti-26-shu-de-zi-jie-gou-xian-xu-bian-li-p/

代码

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

class Solution:
    def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
        # 判断两棵树是不是相同的结构和结点值
        def isSubTree(A, B):
            if not B: return True
            if not A or A.val != B.val: return False
            return isSubTree(A.left, B.left) and isSubTree(A.right, B.right)
        #遍历树中每个节点，判断isSubTree（A，B）
        if not A or not B:
            return False
        if A.val == B.val:
            return isSubTree(A, B)
        else:
            return self.isSubStructure(A.left, B) or self.isSubStructure(A.right, B)

时间复杂度

O(MN)： 其中 M,N分别为树 A 和 树 B 的节点数量；先序遍历树 A 占用 O(M) ，每次调用 isSubTree(A, B) 判断占用O(N) 。

空O(M) ： 当树 A 和树 B 都退化为链表时，递归调用深度最大。当M≤N 时，遍历树 A 与递归判断的总递归深度为 M ；当 M>N 时，最差情况为遍历至树 A 叶子节点，此时总递归深度为 M。