题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
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 pRoot2 == None or pRoot1 == None:
return False
return isSubtree(pRoot1,pRoot2) or isSubtree(pRoot1.left,pRoot2) or isSubtree(pRoot1.right,pRoot2)
def isSubtree(pRoot1,pRoot2):
if pRoot2 == None:
return True
if pRoot1 == None:
return False
if pRoot1.val != pRoot2.val:
return isSubtree(pRoot1.left,pRoot2) or isSubtree(pRoot1.right,pRoot2)
else:
return isSubtree(pRoot1.left,pRoot2.left) and isSubtree(pRoot1.right,pRoot2.right)
解析:
HasSubtree函数返回pRoot2是否是pRoot1、pRoot1.left、pRoot1.right其中之一的子树。
isSubtree递归调用,如果pRoot2走到了空节点,证明pRoot2当前分支是pRoot1的子结构,否则如果pRoot1走到了空节点,证明pRoot2的当前分支不是pRoot1的子结构。
如果pRoot1和pRoot2当前节点的值不相同,则看看
pRoot2是不是
pRoot1左或者右子树的子结构。
否则pRoot1和pRoot2的当前节点的值相同,则继续判断左右子树是否都是pRoot2属于pRoot1的子结构。