[LeetCode] 652. Find Duplicate Subtrees

Given the root of 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 the same node values.

Example 1:

Input: root = [1,2,3,4,null,2,4,null,null,4]
Output: [[2,4],[4]]

Example 2:

Input: root = [2,1,1]
Output: [[1]]

Example 3:

Input: root = [2,2,2,3,null,3,null]
Output: [[2,3],[3]]

Constraints:

• The number of the nodes in the tree will be in the range [1, 10^4]
• -200 <= Node.val <= 200

寻找重复的子树。

给你一棵二叉树的根节点 root ，返回所有 重复的子树 。

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

如果两棵树具有 相同的结构 和 相同的结点值 ，则认为二者是 重复 的。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-duplicate-subtrees
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

请返回一个list，里面包含的是所有重复的子树的根节点。对于同一类的重复子树，你只需要返回其中任意一棵的根结点即可。两棵树重复是指它们具有相同的结构以及相同的结点值。

思路是后序遍历 + 序列化。题眼是子树，那么必然是后序遍历。因为只有先看子树是否相同，把信息传给父节点，在处理父节点的时候我们才知道两棵子树是否相同。

但是如何找相同的子树呢，我们应该是需要一个类似 hashmap 的东西存放子树，这样才能再以后再看到类似的树的结构的时候。这里我们运用到了之前297题的思路，把树的信息序列化。这样我们才能把这个信息当做 key 存到一个 hashmap 里。之后我们把每个子树序列化之后的信息存入 hashmap，并找出重复的子树。

时间O(n)

空间O(n)

Java实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        List<TreeNode> res = new ArrayList<>();
        // corner case
        if (root == null) {
            return res;
        }
        
        // normal case
        helper(root, res, new HashMap<>());
        return res;
    }
    
    private String helper(TreeNode cur, List<TreeNode> res, HashMap<String, Integer> map) {
        if (cur == null) {
            return "#";
        }
        String serial = cur.val + "," + helper(cur.left, res, map) + helper(cur.right, res, map);
        if (map.getOrDefault(serial, 0) == 1) {
            res.add(cur);
        }
        map.put(serial, map.getOrDefault(serial, 0) + 1);
        return serial;
    }
}

 

相关题目

297. Serialize and Deserialize Binary Tree

449. Serialize and Deserialize BST

536. Construct Binary Tree from String

606. Construct String from Binary Tree

652. Find Duplicate Subtrees

LeetCode 题目总结