剑指offer-树的子结构17

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

class Solution:
    def issubTree(self,pRoot1,pRoot2):
        #如果树A空了，树B没空则匹配失败
        if pRoot1==None and pRoot2 is not None:
            return False
        #如果树B空了，匹配完成
        if pRoot2==None:
            return True
        #如果不相等，匹配失败
        if not pRoot1.val==pRoot2.val:
            return False
        #继续匹配
        return self.issubTree(pRoot1.left,pRoot2.left) and self.issubTree(pRoot1.right,pRoot2.right)
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        result=False
        if pRoot1 is not None and pRoot2 is not None:
            #如果发现根节点，则从此根节点开始判断是否匹配
            if pRoot1.val==pRoot2.val:
                result=self.issubTree(pRoot1,pRoot2)
            #如果没有找到，则用左孩子进行匹配
            if not result:
                result=self.HasSubtree(pRoot1.left, pRoot2)
            #如果没有找到，则用右孩子进行匹配
            if not result:
                result=self.HasSubtree(pRoot1.right, pRoot2)
        return result