剑指offer 树的子结构 python


题目描述

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

{8,8,7,9,2,#,#,#,#,4,7},{8,9,2}
返回True

想法一:
使用先序遍历生成两棵树的序列,之后只需要判断序列是否存在即可。

class Solution: def __init__(self): self.Root1_list = [] self.Root2_list = [] # 先序遍历 def Createtreelist(self, root, note): if root is None: return if note is True: self.Root1_list.append(root.val) self.Createtreelist(root.left, True) self.Createtreelist(root.right, True) else: self.Root2_list.append(root.val) self.Createtreelist(root.left, False) self.Createtreelist(root.right, False) def HasSubtree(self, pRoot1, pRoot2): """ :param pRoot1: TreeNode :param pRoot2: TreeNode :return: """ self.Createtreelist(pRoot1, True) self.Createtreelist(pRoot2, False) p = self.Root2_list q = self.Root1_list # 如果为空树直接返回 false if len(p) is 0: return False note2 = 0 for i in range(len(q)): if note2 == len(p): break if q[i] == p[note2]: note2 += 1 else: if q[i] == p[0]: continue note2 = 0 if note2 == len(p): return True return False 

想法二:
通过百度得来的,递归查找,对于递归还是运用的不是很熟练:-)

class TreeNode: def __init__(self, x): self.val = x self.left = None self.right = None class Solution: def SearchSimiler(self, p1, p2): # B树为空,则存在 if p2 is None: return True if p1 is None: return p1 == p2 flag = False if p1.val == p2.val: # 查找该节点的左右节点 flag = self.SearchSimiler(p1.left, p2.left) and self.SearchSimiler(p1.right, p2.right) # 如果flag为True,返回True,否则继续从左节点开始查找,如果还失败,从右节点开始查找 return flag or self.SearchSimiler(p1.left, p2) or self.SearchSimiler(p1.right, p2) def HasSubtree(self, pRoot1, pRoot2): """ :param pRoot1: TreeNode :param pRoot2: TreeNode :return: """ # write code here if pRoot1 is None or pRoot2 is None: return False return self.SearchSimiler(pRoot1, pRoot2) 

最后

刷过的LeetCode源码或剑指offer放在Github上了,希望喜欢或者觉得有用的朋友点个star或者follow。
有任何问题可以在下面评论或者通过私信或联系方式找我。
联系方式
QQ:791034063
Wechat:liuyuhang791034063
CSDN:https://blog.csdn.net/Sun_White_Boy
Github:https://github.com/liuyuhang791034063


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM