from pickle import NONE
# class TreeNode:
# 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 pRoot1==None or pRoot2==None:
return False
if self.Judge(pRoot1,pRoot2):
return True
return self.HasSubtree(pRoot1.left,pRoot2) or self.HasSubtree(pRoot1.right,pRoot2)
def Judge(self,root1,root2):
# 1空2空为真 1空2不空为假 1不空2空为真 1不空2不空则继续判断左右部分
if root1==None and root2==None:
return True
if root1==None and root2!=None:
return False
if root1 != None and root2==None:
return True
# 需要再判断一下根节点的值是不是相等
if root1.val != root2.val:
return False
return self.Judge(root1.left,root2.left) and self.Judge(root1.right,root2.right)
- 主函数判断根节点的A树是否有B树结构,没有的话,递归判断A树的子树节点
- 用一个judge函数判断当前根节点的A树中是否包含子树B结构
- 主要注意点在于判断条件