重复子树的识别：揭开LeetCode 652题的奥秘

2024-01-29 00:52:57

作为一名技术博客创作专家，我很乐意与大家分享我对 LeetCode 652 题的独到见解。这道题目的主要目的是识别二叉树中的重复子树。重复子树是指在二叉树中存在结构和值都相同的子树。为了解决这个问题，我将使用一种有效且通用的算法，即哈希表结合深度优先搜索（DFS）。

首先，我们需要创建一个哈希表来存储已经遍历过的子树。在 DFS 的过程中，我们将对每个子树进行序列化，并将其作为哈希表的键。如果某个子树的序列化结果已经在哈希表中，则说明这是一个重复的子树。

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

public class FindDuplicateSubtrees {

    private HashMap<String, Integer> subtreeMap = new HashMap<>();
    private List<TreeNode> duplicateSubtrees = new ArrayList<>();

    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        serialize(root);
        return duplicateSubtrees;
    }

    private String serialize(TreeNode root) {
        if (root == null) {
            return "";
        }
        String left = serialize(root.left);
        String right = serialize(root.right);
        String current = root.val + "," + left + "," + right;
        int count = subtreeMap.getOrDefault(current, 0) + 1;
        subtreeMap.put(current, count);
        if (count == 2) {
            duplicateSubtrees.add(root);
        }
        return current;
    }
}