题解 | #树的子结构#

一、思路

采用递归的方法，先声明一个dfs函数，用于比较以node2为根节点的子树是不是以node1为根节点的子树，具体见代码注释。

#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param pRoot1 TreeNode类 
# @param pRoot2 TreeNode类 
# @return bool布尔型
#
class Solution:
    def HasSubtree(self , pRoot1: TreeNode, pRoot2: TreeNode) -> bool:
        # write code here
        #特判，当两个树有一个为空，都不会构成子树关系
        if not pRoot2 or not pRoot1:
            return False 
        #深度优先搜索，进行比较，看当前节点为根节点树是否能构成子集关系
        def dfs(node1,node2):
            if not node2:
                return True 
            elif not node1:
                return False
            elif node1.val != node2.val:
                return False
            return dfs(node1.left,node2.left) and dfs(node1.right,node2.right)
        #递归调用，先从根节点开始搜索，再从左右子树去搜索比较
        return dfs(pRoot1,pRoot2) or self.HasSubtree(pRoot1.left, pRoot2) or self.HasSubtree(pRoot1.right, pRoot2)