给定一棵二叉树 root,返回所有重复的子树。
对于同一类的重复子树,你只需要返回其中任意一棵的根结点即可。
如果两棵树具有相同的结构和相同的结点值,则它们是重复的。
示例1
输入:root = [1,2,3,4,null,2,4,null,null,4] 输出:[[2,4],[4]]
- 由于题目要求我们把所有的重复子树都找出来 那么重复子树的子树也应该在其中 所以我们应该从叶子节点开始找起
- 因此使用后序查找方式是更为合适的
- 为了标记唯一的树类型 对于None的子节点 我们需要用一个特殊符号“#”来标记
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def findDuplicateSubtrees(self, root: Optional[TreeNode]) -> List[Optional[TreeNode]]:
if not root:return []
res = []
counter = collections.Counter()
def traverse(root):
if not root:return '#'
left = traverse(root.left)
right = traverse(root.right)
chain = left + ',' + right + ',' + str(root.val)
counter[chain] += 1
if counter[chain] == 2:
res.append(root)
return chain
traverse(root)
return res
Counter作为dict 字典的子类,其优点在于,如果在 Counter 中查找一个不存在的元素,不会产生异常,而是会返回 0,这正是我们需要的。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-duplicate-subtrees