题目:
剑指 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为树的深度。