题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路:
递归,如果当前节点的val一样,则用issubTree判断左右节点是否一样,一直递归到pRoot2位空代表是包含,否则返回False。
如果不一样,则递归左右子树,因为or返回第一个为真的对象,所以不用担心内存过大。
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
def HasSubtree(self, pRoot1, pRoot2):
if not pRoot2 or not pRoot1:
return False
return self.issubtree(pRoot1,pRoot2) or self.HasSubtree(pRoot1.left,pRoot2) or self.HasSubtree(pRoot1.right, pRoot2) #返回第一个为真的值,剩下的不在计算。
def issubtree(self,pRoot1, pRoot2):
if pRoot2 == None: return True #这一行与下一行不能对换,因为存在proot1和2都是空的情况
if pRoot1 == None: return False
if pRoot1.val == pRoot2.val:
return self.issubtree(pRoot1.left, pRoot2.left) and self.issubtree(pRoot1.right, pRoot2.right)