LC:剑指 Offer 26. 树的子结构

题目：

剑指 Offer 26. 树的子结构.

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

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

链接：https://leetcode-cn.com/leetbook/read/illustrate-lcof/e28c7j/

示例：

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

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

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

思路：

首先要区分子树和子结构的概念：

• 子树：若 B 是 A 的子树，则 A包含B的所有结点，并且只要包含了 B 的一个结点，还得包含这个结点下的所有节点。
• 子结构：若 B 是 A 的子结构，则 A 包含 B 的所有结点，但子结构 B 仅是 A 树在该结点下的一部分，子结构B不需要包含这个结点下A的所有结点。

本题为求是否为子结构，并且空树不作为树的子结构，所以若B为空可以直接返回False。

本题采用双重递归求解。第一层isSubStructure，用于遍历A树，寻找与B相同的子结构；第二层findSubStructure，用于判断B树与A的子结构是否相等，递归判断B的每一个结点。

代码：

# 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:
        if not B or not A: return False
        return self.findSubStructure(A, B) or self.isSubStructure(A.left, B) or self.isSubStructure(A.right, B)

    def findSubStructure(self, rootA, rootB):
        if not rootB: return True
        if not rootA: return False
        if rootA.val != rootB.val: return False
        return self.findSubStructure(rootA.left, rootB.left) and self.findSubStructure(rootA.right, rootB.right)

时间复杂度O(mn)，空间复杂度O(n). m为树的深度。