给定一棵二叉树 root,返回所有重复的子树。
对于同一类的重复子树,你只需要返回其中任意一棵的根结点即可。
如果两棵树具有相同的结构和相同的结点值,则它们是重复的。
示例1
输入:root = [1,2,3,4,null,2,4,null,null,4] 输出:[[2,4],[4]]
由于题目要求我们把所有的重复子树都找出来 那么重复子树的子树也应该在其中 所以我们应该从叶子节点开始找起
因此使用后序查找方式是更为合适的
为了标记唯一的树类型 对于None的子节点 我们需要用一个特殊符号“#”来标记
class Solution:
def __init__(self):
self.rec = {}
self.l = []
def findDuplicateSubtrees(self, root: Optional[TreeNode]) -> List[Optional[TreeNode]]:
rec = {}
lst = []
def postorder(root):
if root is None:
return '#'
l = postorder(root.left)
r = postorder(root.right)
s = '_'.join((l , r, str(root.val)))
# rec
if s not in rec:
rec[s] = 1
elif s in rec and rec[s]==1:
lst.append(root)
rec[s] += 1
elif s in rec and rec[s] !=1:
rec[s] +=1
return s
postorder(root)
return lst
看不懂的朋友,可以去看看这个链接有详细的解释https://blog.csdn.net/weixin_48276298/article/details/117914419
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-duplicate-subtrees