问题:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解决思路:大致可以分成两步来实现
第一步:比较根节点,如果根节点相同继续比较剩下的叶子节点,如果不同,则需要继续向下遍历(递归实现)第一课树的其余节点
第二步:根节点相同的情况下,对叶子节点进行比较(递归实现对叶子的遍历)
1 # -*- coding:utf-8 -*- 2 # class TreeNode: 3 # def __init__(self, x): 4 # self.val = x 5 # self.left = None 6 # self.right = None 7 class Solution: 8 def HasSubtree(self, pRoot1, pRoot2): 9 # write code here 10 result = False # 标志位 11 if pRoot1 and pRoot2: #边界条件,都不为空 12 if pRoot1.val == pRoot2.val: 13 result = self.equal(pRoot1, pRoot2) #返回结果如果是True,则不需要继续向下遍历,如果是False则需要继续向下遍历第一棵树 14 if not result: 15 result = self.HasSubtree(pRoot1.left, pRoot2) 16 if not result: 17 result = self.HasSubtree(pRoot1.right, pRoot2) 18 return result 19 20 def equal(self, pRootA, pRootB): 21 #三个跳出递归的条件,B结束(相等True),A结束(False),B的值不等于A的值(False) 22 if not pRootB: 23 return True 24 if not pRootA: 25 return False 26 if pRootA.val != pRootB.val: 27 return False 28 #跳不出递归,则继续向下比较 29 return self.equal(pRootA.left, pRootB.left) and self.equal(pRootA.right, pRootB.right)