剑指offer 之树的子结构

"""
问题：寻找二叉树的子结构
    输入两棵二叉树A，B，判断B是不是A的子结构。
    （ps：我们约定空树不是任意一个树的子结构）
思路：
    1.如果B树是A树的子树,会有三种情况
        a.从A的根节点开始相同
        b.从A的左子树开始相同
        c.从A的右子树开始相同
    2.利用递归的思想分别讨论,有一种情况吻合就说明符合题意的情况
    
"""
# -*- coding:utf-8 -*-
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        if not pRoot1 or not pRoot2:
            return False
        # a.从A的根节点开始相同
        res1 = self.is_subtree(pRoot1, pRoot2)
        # b.从A的左子树开始相同
        res2 = self.is_subtree(pRoot1.left, pRoot2)
        # c.从A的右子树开始相同
        res3 = self.is_subtree(pRoot1.right, pRoot2)
        return res1 or res2 or res3

    def is_subtree(self, root1, root2):
        # 如果B树没有元素了,直接返回False
        if not root2:
            return True
        # 如果A树没有元素了,或者A和B元素的值不相等,直接返回False
        if not root1 or root1.val != root2.val:
            return False
        res1 = self.is_subtree(root1.left, root2.left)
        res2 = self.is_subtree(root1.right, root2.right)
        return res1 and res2