树的子结构
解题思路:
若树 B 是树 A的子结构,则子结构的根节点可能为树 A 的任意一个节点。因此,判断树 B 是否是树 A 的子结构,需完成以下两步工作:
-
先序遍历树 A 中的每个节点 nA;(对应函数 isSubStructure(A, B))。
-
判断树 A 中 以 nA为根节点的子树 是否包含树 B 。(对应函数 recur(A, B))。
算法流程:
名词规定:树 A的根节点记作 节点 A ,树 B 的根节点称为 节点 B。
recur(A, B) 函数:
1.终止条件:
1.当节点 B 为空:说明树 BBB 已匹配完成(越过叶子节点),因此返回 truetruetrue ;
2.当节点 A 为空:说明已经越过树 AAA 叶子节点,即匹配失败,返回 falsefalsefalse ;
3.当节点 A 和 B的值不同:说明匹配失败,返回 falsefalsefalse ;
2.返回值:- 判断 A 和 B 的左子节点是否相等,即 recur(A.left, B.left) ;
- 判断 A 和 B 的右子节点是否相等,即 recur(A.right, B.right) ;
isSubStructure(A, B) 函数:
-
特例处理: 当 树 A 为空 或 树 B 为空 时,直接返回 falsefalsefalse ;
-
返回值: 若树 B是树 A 的子结构,则必满足以下三种情况之一,因此用或 || 连接;
1.以 节点 A 为根节点的子树 包含树 BBB,对应 recur(A, B);
2.树 B 是 树 A 左子树 的子结构,对应 isSubStructure(A.left, B);
3.树 B 是 树 A 右子树 的子结构,对应 isSubStructure(A.right, B);
代码:
class Solution:
def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
def recur(A, B):
if not B: return True
if not A or A.val != B.val: return False
return recur(A.left, B.left) and recur(A.right, B.right)
return bool(A and B) and (recur(A, B) or self.isSubStructure(A.left, B) or self.isSubStructure(A.right, B))