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)
京公网安备 11010502036488号