项目有个需求,蛮难表达清楚的。就是1对多的求并集合并,例如一个Map如下
A=>1,2
B=>3,4,6
C=>2,5
D=>6
E=>6,7
F=>8
A和C共同含有2,那么需要合并,最后合并成
A,C=>1,2,5
B,D,E=>3,4,6,7
F=>8
思路大概是两次遍历Map,如果发现结果有交集,合并。然后再重新遍历,直到所有的都没交集为止。用Java的话,在遍历Map的时候无法对改Map进行操作,所以用了递归。
代码如下:
public void merge(Map<String, List<Long>> tmpMap) {
Map<String, List<Long>> map = new HashMap<String, List<Long>>(tmpMap);
for (String key1 : map.keySet()) {
for (String key2 : map.keySet()) {
if (key1.equals(key2)) continue;
List<Long> tmp1 = new ArrayList<Long>(map.get(key1));
List<Long> tmp2 = new ArrayList<Long>(map.get(key2));
tmp1.retainAll(tmp2);
if (tmp1.size() > 0) {
Set<Long> a = new HashSet<Long>();
a.addAll(tmpMap.get(key1));
a.addAll(tmpMap.get(key2));
tmpMap.remove(key1);
tmpMap.remove(key2);
tmpMap.put(key1 + "," + key2, new ArrayList<Long>(a));
merge(tmpMap);
return;
}
}
}
}
方法传的是引用,所以直接把tmpMap给改了。写完代码感觉性能应该很差,不过先用着,不知道有没有什么现成的方法。以后找找看。