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

桃扇骨 2023-07-21 05:57 7阅读 0赞

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