今天碰到同事在群里問怎么把一個Map根據value進行分組,當時我的第一想法是將value和key互換,
多個value對應的不同key作為集合,代碼如下:
1
2
Map<String,Integer>SataMap=new HashMap<>();
final Map<Integer, List<String>> groupMap = new HashMap<>();
SataMap.put("A",1);
SataMap.put("b",3);
SataMap.put("c",2);
SataMap.put("D",4);
SataMap.put("e",2);
SataMap.put("f",4);
SataMap.put("G",3);
SataMap.put("h",2);
System.out.println(result);
SataMap.forEach(new BiConsumer<String, Integer>() {
@Override
public void accept(String key, Integer value) {
if(groupMap.containsKey(value)){
groupMap.get(value).add(key);
}else{
List<String> values = new ArrayList<>();
values.add(key);
groupMap.put(value, values);
}
}
});
結果很快被打臉了,同事找到一個更好的方法
Map<Integer, List<Map.Entry<String,Integer>>>result= SataMap.entrySet().stream().collect(Collectors.groupingBy(c -> c.getValue()));
1
將Map轉換為一個entry集合,然后再用集合分組的方式,就很簡單的實現這個功能了