分享一个自己写拼有趣最近的高频店面题 希望大佬指教

最近看到大佬们发了一拼有趣最近的店面题 找相似的images
题目链接
https://www.1point3acres.com/bbs ... 6orderby%3Ddateline

看到之前也有人发说是用union find比较好解。自己用java写了一个试试。希望大佬们能指教

1.     public static List<List<String>> findSim(Map<String, List<String>> map) {
   
2.         Map<String, String> parents = new HashMap<>()
   
3.         Map<String, TreeSet<String>> unions = new HashMap<>();
   
4.         
   
5.         for (String key : map.keySet()) {
   
6.             parents.put(key, key)
   
7.             for (String v : map.get(key)) {
   
8.                 parents.put(v,v)
   
9.             }
   
10.         }
    
11.         
    
12.         for (String key : map.keySet()) {
    
13.             String p = find(key, parents);
    
14.             for (String v : map.get(key)) {
    
15.                 parents.put(find(v, parents), p);
    
16.             }
    
17.         }
    
18.    
    
19.         for (String key : map.keySet()) {
    
20.             String p = find(key, parents);
    
21.             if (!unions.containsKey(p)) {
    
22.                 unions.put(p, new TreeSet<String>());
    
23.             }
    
24.             unions.get(p).add(p);
    
25.             for (String v : map.get(key)) {
    
26.                 unions.get(p).add(v);
    
27.             }
    
28.         }
    
29.         List<List<String>> res = new ArrayList<>();
    
30.         for (String key : unions.keySet()) {
    
31.             List<String> cur = new ArrayList<>();
    
32.             for (String val : unions.get(key)) {
    
33.                 cur.add(val);
    
34.             }
    
35.             res.add(cur);
    
36.         }
    
37.         return res;
    
38.     }

复制代码