LeetCode 652: 寻找重复的子树 Find Duplicate Subtrees

题目:

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

两棵树重复是指它们具有相同的结构以及相同的结点值。

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.

示例 1：

        1
       / \
      2   3
     /   / \
    4   2   4
       /
      4

下面是两个重复的子树：

      2
     /
    4

和

    4

因此，你需要以列表的形式返回上述重复子树的根结点。

Therefore, you need to return above trees' root in the form of a list.

解题思路:

这就是一道考察二叉树遍历的题, 遍历的时候序列化作为 String 类型存入 Map, 若其为第二次出现即将该结点加入数组.

代码:

这里以后序遍历为例, 需要注意叶子结点应当规定一个特殊字符作为替代 null 结点, 这里用的是 '#'

Java:

class Solution {
    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        List<TreeNode> list = new ArrayList<>();//待返回数组
        if (root == null) return list;
        Map<String, Integer> map = new HashMap<>();//哈希映射查找重覆子结点
        traverse(root, map, list, "");
        return list;
    }
    //后序遍历
    private String traverse(TreeNode node, Map<String, Integer> map, List<TreeNode> list, String tree) {
        if (node == null) return tree + "#";
        String left = traverse(node.left, map, list, tree);//递归左子孩子
        String right = traverse(node.right, map, list, tree);//递归右子孩子
        tree = tree + node.val + left + right;//每个子树路径

        map.put(tree, map.getOrDefault(tree, 0) + 1);//存储每个子树路径
        if (map.get(tree) == 2) list.add(node);//第二次出现相同路径时存储该结点
        return tree;
    }
}

Python:

class Solution:
    def findDuplicateSubtrees(self, root: TreeNode) -> List[TreeNode]:
        res = [] # 待返回数组
        if not root: return res
        hash_map = {} # 哈希映射查找重覆子结点
        self.traverse(root, hash_map, res, '')
        return res
    # 后序遍历
    def traverse(self, node: TreeNode, hash_map, res, tree):
        if not node: return tree + '#'
        left = self.traverse(node.left, hash_map, res, tree) # 递归左子孩子
        right = self.traverse(node.right, hash_map, res, tree) # 递归右子孩子
        tree = tree + str(node.val) + left + right # 每个子树路径
        if tree in hash_map.keys(): # 存储每个子树路径
            hash_map[tree] += 1
        else:
            hash_map[tree] = 1
        if hash_map[tree] == 2: # 第二次出现相同路径时存储该结点
            res.append(node)
        return tree

欢迎关注微信公众号: 爱写Bug