树的子结构

题目描述

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

# -*- 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
        result = False
        if pRoot1 != None and pRoot2 != None:
            # 先找到A的根节点是否与B相同
            # 直到找到A中任一父节点与B相同，就接着比较下面的子节点
            # 若不同则接着搜索左右子树，直到A树结束
            if pRoot1.val == pRoot2.val:
                result = self.same(pRoot1, pRoot2)
            # 若根节点不同则拿A的左子树与B比较
            # 搜索左子树
            if not result:
                result = self.HasSubtree(pRoot1.left, pRoot2)
            # 若仍不同则将A的右子树与B比较
            # 若仍未找到则A不包含B
            # 搜索右子树
            if not result:
                result = self.HasSubtree(pRoot1.right, pRoot2)
        return result
    # 这里比较根节点相同的情况
    def same(self, pRoot1, pRoot2):
        # 若B的子节点为空，则A包含B
        if pRoot2 == None:
            return True
        # 若A的子节点为空，则A不包含B
        if pRoot1 == None:
            return False
        # 若接下来的值不同，则A不包含B
        if pRoot1.val != pRoot2.val:
            return False
        # 接着递归判断A与B的左右子树是否相同
        return self.same(pRoot1.left, pRoot2.left) and self.same(pRoot1.right, pRoot2.right)