17. 树的子结构

题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)


思路
我们约定空树不是任意一个树的子结构,所以只有当A、B均不为空时才判断B树是否为A的子树、子结构。
如果B是A的子结构,则B的根节点一定在A中出现,首先在A中找到B的根节点所在,然后判断A树中这个节点下是否出现B的结构。


python代码实现

# -*- 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(pRoot1 and pRoot2):
            return self.isSubtree(pRoot1,pRoot2) or self.HasSubtree(pRoot1.left,pRoot2) or self.HasSubtree(pRoot1.right,pRoot2)
        else:
            return False

    def isSubtree(self, pRoot1, pRoot2):
        if(not pRoot2):
            return True
        elif(not pRoot1 or pRoot1.val != pRoot2.val):
            return False
        else:
            return self.isSubtree(pRoot1.left,pRoot2.left) and self.isSubtree(pRoot1.right,pRoot2.right)