寻找重复的子树

/*
 * 题目思路：
 * 同样地，我们需要找到二叉树中的重复子树。我们使用Java Stream API来实现序列化子树并检测重复。
 * 使用递归和Collector将子树的序列化表示聚合到一个列表中，然后利用Collectors.groupingBy来分组计算相同子树的出现次数。
 * 最后筛选出重复的子树。
 */
 
import java.util.*;
import java.util.stream.*;
 
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}
 
public class Solution {
    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        return findDuplicates(root).values().stream()
                .filter(list -> list.size() > 1)
                .map(list -> list.get(0))
                .collect(Collectors.toList());
    }
    
    private Map<String, List<TreeNode>> findDuplicates(TreeNode node) {
        if (node == null) return new HashMap<>();
        
        Map<String, List<TreeNode>> leftMap = findDuplicates(node.left);
        Map<String, List<TreeNode>> rightMap = findDuplicates(node.right);
        
        String serial = node.val + "," + leftMap.keySet().stream().findFirst().orElse("#") + "," + rightMap.keySet().stream().findFirst().orElse("#");
        
        Map<String, List<TreeNode>> currentMap = new HashMap<>();
        currentMap.put(serial, new ArrayList<>(Arrays.asList(node)));
        
        leftMap.forEach((key, value) -> currentMap.merge(key, value, (l1, l2) -> { l1.addAll(l2); return l1; }));
        rightMap.forEach((key, value) -> currentMap.merge(key, value, (l1, l2) -> { l1.addAll(l2); return l1; }));
        
        return currentMap;
    }
}