Java实现 LeetCode 652 寻找重复的子树（两个map的DFS）

652. 寻找重复的子树

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

两棵树重复是指它们具有相同的结构以及相同的结点值。

示例 1：

 1
   / \
  2   3
 /   / \
4   2   4
   /
  4

下面是两个重复的子树：

      2
     /
    4
和

    4

因此，你需要以列表的形式返回上述重复子树的根结点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
  private int treeId = 100;
    //这个用来保存树得，用一个id去替换树
    private Map<String, Integer> tree = new HashMap<>();
    //下面这个是用上面树换的id来保存次数得，只有两次得时候才会加入到下面得list里面
    private Map<Integer, Integer> count = new HashMap<>();
    private List<TreeNode> res;

    public List<TreeNode> findDuplicateSubtrees(TreeNode root) {
        res = new ArrayList<>();
        this.dfs(root);
        return res;
    }

    private int dfs(TreeNode node){
        if(node == null) return 0;

        String content = new StringBuilder().append(node.val).append(",").append(this.dfs(node.left)).append(",").append(this.dfs(node.right)).toString();

        Integer id = tree.get(content);
        if(id == null) {
            id = treeId++;
            tree.put(content, id);
        }

        int num = count.getOrDefault(id, 0);
        count.put(id, ++num);

        if(num == 2) {
            res.add(node);
        }
        return id;
    }

}