我的地图看起来像这样
private Map<String, LinkedHashSet<String>> map = new HashMap<>();
在传统方法中,我可以使用键检查为地图添加值,如下所示
public void addEdge(String node1, String node2) {
LinkedHashSet<String> adjacent = map.get(node1);
if (adjacent == null) {
adjacent = new LinkedHashSet();
map.put(node1, adjacent);
}
adjacent.add(node2);
}
使用java 8,我可以做这样的事情,我也可以得到相同的输出。
map.compute(node1, (k,v)-> {
if(v==null) {
v=new LinkedHashSet<>();
}
v.add(node2);
return v;
});
有没有更好的方法来处理java 8?
答案 0 :(得分:5)
使用
map.computeIfAbsent(node1, k -> new LinkedHashSet<>()).add(node2);
如果地图中已找到node1
,则相当于:
map.get(node1).add(node2);
如果地图中尚未显示node1
,则相当于:
map.put(node1, new LinkedHashSet<>()).add(node2);
这正是您正在寻找的内容,甚至被描述为documentation中的用例。