leetcode 652 寻找重复的子树(python)

给定一棵二叉树 root，返回所有重复的子树。

对于同一类的重复子树，你只需要返回其中任意一棵的根结点即可。

如果两棵树具有相同的结构和相同的结点值，则它们是重复的。

示例1

• 由于题目要求我们把所有的重复子树都找出来 那么重复子树的子树也应该在其中 所以我们应该从叶子节点开始找起
• 因此使用后序查找方式是更为合适的
• 为了标记唯一的树类型 对于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，这正是我们需要的。