【Leetcode】652. Find Duplicate Subtrees

题目地址：

https://leetcode.com/problems/find-duplicate-subtrees/

给定一棵二叉树，求其所有的出现多次的子树，返回这些子树的根（每种类型的子树返回任意的其中一个根即可）。

思路是DFS + 序列化。在DFS到某个节点的时候，就将该节点为根的子树进行序列化（进行先序遍历序列化。当然其他序的序列化也是可以的），然后存入HashMap中。如果某个序列化的字符串出现了多次，则说明存在多个形状相同的树。代码如下：

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Solution {
    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        List<TreeNode> res = new ArrayList<>();
        Map<String, List<TreeNode>> map = new HashMap<>();
        dfs(root, map);
        for (Map.Entry<String, List<TreeNode>> entry : map.entrySet()) {
        	// 出现了重复，则加入答案
            if (entry.getValue().size() > 1) {
                res.add(entry.getValue().get(1));
            }
        }
        
        return res;
    }
    
    private String dfs(TreeNode root, Map<String, List<TreeNode>> map) {
    	// 空树的序列化是"#"，但空树不必put进哈希表里
        if (root == null) {
            return "#";
        }
        
        // root子树的序列化就是树根 + 空格 + 左子树序列化 + 空格 + 右子树序列化
        String s = String.valueOf(root.val) + ' ' + dfs(root.left, map) + ' ' + dfs(root.right, map);
        map.putIfAbsent(s, new ArrayList<>());
        map.get(s).add(root);
        return s;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    
    public TreeNode(int val) {
        this.val = val;
    }
}

时间复杂度$O(n\log n)$（其实是由Master定理得到，$T(n)=2T(\frac{n}{2})+O(n)$。如果树非常不平衡的话，可能会退化到$O(n^2)$），空间$O(n)$。