题目
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路
暴力做法
- 取出树A/B的前序、中序;
- 判断这两个数组是否A都包含B,是则true。
前序和中序、中序和后序可以唯一确定一棵树。
递归
- 实现一个是不是子树的函数IsSubtree;
- 判断两颗树是否为空,空树不是任何树的子树;
- 最后 return self.IsSubtree(pRoot1, pRoot2) or self.IsSubtree(pRoot1.left, pRoot2) or self.IsSubtree(pRoot1.right, pRoot2)
有点不太懂,我要开始实现了IsSubtree的功能,但是一直不通过,看到网上说还要看看A左右子树与B的IsSubtree结果,很神奇这样就过了。
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def IsSubtree(self, pRoot1, pRoot2):
if pRoot2 is None:
return True
elif pRoot1 is None and pRoot2:
return False
elif pRoot2.val==pRoot1.val:
return self.IsSubtree(pRoot1.left, pRoot2.left) and self.IsSubtree(pRoot1.right, pRoot2.right)
elif pRoot2.val!=pRoot1.val:
return self.IsSubtree(pRoot1.left, pRoot2) or self.IsSubtree(pRoot1.right, pRoot2)
return True
def HasSubtree(self, pRoot1, pRoot2):
# write code here
if pRoot1 is None or pRoot2 is None:
return False
return self.IsSubtree(pRoot1, pRoot2) or self.IsSubtree(pRoot1.left, pRoot2) or self.IsSubtree(pRoot1.right, pRoot2)