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

}