力扣 652. 寻找重复的子树

题目来源：https://leetcode.cn/problems/find-duplicate-subtrees/

大致题意：
给一个二叉树，返回所有结构相同且节点值相同的子树

思路

可以通过给二叉树节点编码，然后中序遍历的方式解题

对每颗子树进行编码后，使用哈希表存下已有的编码和编码对应的子树根节点，若对应编码已经在哈希表中，则表明有重复子树，将哈希表中的子树根节点放入答案集合中。因为相同的子树可能出现多次，所以采用哈希表存答案集合。

具体的编码方式为：

1. 使用一个全局变量 id 表示当前应该给未遍历节点的 id
2. 首先通过中序遍历获得当前节点左右子树的编码值，然后加上当前节点值组成编码：左子树编码 + 当前节点值 + 右子树编码
3. 判断当前编码是否已经在哈希表中，若已有，则将哈希表中该编码对应的节点加入答案集合，同时获取对应节点的编码作为当前节点的 id；否则，将全局变量 id 赋给当前节点作为 id，同时更新 id

具体看代码：

public class FindDuplicateSubtrees {

    Set<TreeNode> set = new HashSet<>();    // 重复子树的根节点集合
    Map<String, Integer> idMap = new HashMap<>();   // 存编码对应的节点 id
    Map<String, TreeNode> nodeMap = new HashMap<>(); // 存编码对应的节点
    int id = 1; // 全局 id
    
    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        dfs(root);
        return new ArrayList<>(set);
    }

    public int dfs(TreeNode node) {
        // 若节点为空，返回 0
        if (node == null) {
            return 0;
        }
        // 拼接编码
        String key = node.val + "" + dfs(node.left) + dfs(node.right);
        // 若该编码在哈希表中存在，则将对应节点放入答案集合并将对应节点的 id 赋给当前节点
        if (idMap.containsKey(key)) {
            set.add(nodeMap.get(key));
            return idMap.get(key);
        } else {    // 否则，将全局 id 赋给当前节点，并更新哈希表和 id
            idMap.put(key, id++);
            nodeMap.put(key, node);
            return id;
        } 
    }
    

    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;
        }
    }
}