[leetcode]652. Find Duplicate Subtrees(Python)

最新推荐文章于 2020-09-10 11:44:22 发布

La_Ragazza

最新推荐文章于 2020-09-10 11:44:22 发布

阅读量298

点赞数

分类专栏： leetcode 文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/La_Ragazza/article/details/105065274

版权

leetcode 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

这篇博客介绍了LeetCode中的第652题——查找重复子树。题目要求在给定的二叉树中找到所有结构和节点值都相同的重复子树，并返回它们的根节点。解题思路是利用先序遍历或后序遍历来保存子树结构，并使用哈希表进行查找，以避免重复。文章提到了一个参考链接，并且指出可以使用`collections.defaultdict`来优化计数过程。

摘要由CSDN通过智能技术生成

[leetcode]652. Find Duplicate Subtrees

题目描述

Category：Mediudm Tree

Given a binary tree, return all duplicate subtrees. For each kind of duplicate subtrees, you only need to return the root node of any one of them.
Two trees are duplicate if they have the same structure with same node values.

题目理解

给定一棵二叉树，返回所有重复的子树。对于每种类型的重复子树，返回其中任何一种的根节点。
重复子树定义：如果两个树具有相同的结构和相同的节点值，则它们是重复的。

解题思路

结合树和hash
每一棵子树，都能把它的结构使用先序遍历或者后序遍历保存下来，然后把这个结构保存在hash表格里面，这样当我们下次再遇到这个树的结构的时候，就能很容易查表得知，然后放到结果res中。
思路借鉴：https://blog.csdn.net/fuxuemingzhu/article/details/81053453

class Solution:
    # Runtime: 56ms 85.68%  MemoryUsage: 21.4MB 40.00%
    def findDuplicateSubtrees(self, root: TreeNode) -> List[TreeNode]:
        res = []
        m = collections.defaultdict(int)
        self.dfs(root, m, res)
        return res

    def dfs(self, root, m, res):
        if not root:
            return '#'
        path = str(root.val) + ',' + self.dfs(root.left, m, res) + ',' + self.dfs(root.right, m, res)
        # print(path)
        if m[path] == 1:
            res.append(root)
        m[path] += 1
        return path

打印中间结果：

4,#,#
2,4,#,#,#
4,#,#
2,4,#,#,#
4,#,#
3,2,4,#,#,#,4,#,#
1,2,4,#,#,#,3,2,4,#,#,#,4,#,#

补：collections.defaultdict
字典的子类，提供了一个工厂函数，为字典查询提供一个默认值
返回一个新的类似字典的对象。 defaultdict 是内置 dict 类的子类。它重载了一个方法并添加了一个可写的实例变量。其余的功能与 dict 类相同，此处不再重复说明。
设置 default_factory 为 int ，使 defaultdict 在计数方面发挥好的作用
原文地址(python3.7)