题目描述:
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B:
4
/
1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
思路:树是天然的递归结构,这里可以考虑两次递归的用法
首先是isSubStructure函数,用来遍历树A,查询B的根节点是否在A中
第二个递归是自己写一个递归函数ischild,用来判断以当前A节点为根节点的树结构中左右子树是否每个节点都满足和B树相等的关系。
整个的过程就是,先运行isSubStructure函数,参数传入AB,就是AB的根节点,判断当前两个根节点是否相等,不相等就return False,继续遍历A树的左子树和B是否相等并上A的右子树和B是否相等。如果找到对应的根节点了,就遍历ischild函数,判断当前节点的左右孩子节点是否满足条件
剩下的思路解析在代码中
代码:
# 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:
#先定义另一个递归函数,用来判断B是否为A的子结构
def child(A,B):
#如果没有B节点了,说明B节点已经超越叶子节点,且当前递归没有报错,那就返回True
if not B:return True
#如果没有A节点了或者AB节点不相等,那就return False
if not A or A.val!=B.val:return False
#上述情况都没有发生,那就return 左右子节点分别再进行递归
return ischild(A.left,B.left) and ischild(A.right,B.right)
#大递归函数中,首先是判断有没有AB子树
if not A or not B:return False
#首先调用小递归函数,判断都处于A,B根节点的时候是否符合
if ischild(A,B):return True
#不符合就继续递归
return self.isSubStructure(A.left,B) or self.isSubStructure(A.right,B)