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)