当Set为Hashmap的值时，将元素添加到Set

Question

这是我的地图

HashMap<String, Set<String>> hmap = new HashMap<>();

我想从我的Map检索一个Set并向其中添加一个元素。

Set<String> val = hmap.get(key);
val.add(newElement);    //NPE
hmap.put(key, val);

但我在val.add行上得到NullPointerException

这可能是微不足道的，但我没有看到它..

我错过了什么？

Answer 1

请尝试以下选项：

HashMap<String, Set<String>> hmap = new HashMap<>();
Set<String> val = hmap.get(key);
if (val == null) {
    val = new HashSet<>();
    val.add(newElement);
    hmap.put(key, val);
}
else {
    val.add(newElement);
}

Answer 2

如果您正在编写修改给定键的映射条目的代码，那么首先检查该特定键是否存在值是一个好主意。

例：

if (hmap.contains(key) == false) 
{
    // This will prevent the null reference issue
    hmap.put(key, new HashSet<String>());
}
Set<String> val = hmap.get(key);
val.add(newElement);

另外，作为旁注：

hmap.put(key, val);

那条线不是必需的。 “val”是对地图中现有集合的引用。 修改它将修改映射到key的值，因此您不需要调用put。 它已经存在并且已经更新。

Answer 3

您可以使用Map.computeIfAbsent() ，如果地图中不存在该键，则会添加该值。

请注意，该方法返回与指定键关联的当前（现有或计算）值。 因此，您可以使用返回的对象在Set对象中添加String对象：

String key = "a key";
Set<String> val = hmap.computeIfAbsent(key, k-> new HashSet<>()); // create the hashset and associate it to the key if key not present
val.add("a value");

Answer 4

这就是实际发生的事情

Map<String, Set<String>> map = new HashMap<String, Set<String>>();
map.put("1_a", new HashSet<String>());

Set<String> set1 = map.get("1_a");
set1.add("a");

System.out.println(set1);

Set<String> set2 = map.get("1_b");  // Not present
set2.add("a");   // NPE - set2 will be null.

你可以添加@Stephano提到的支票，它总是很方便。

Answer 5

为了缩短你，你也可以这样做：

HashMap<String, Set<String>> hmap = new HashMap<>();
String key = "key";
String newElement = "ele";

if (!hmap.containsKey(key)) {
    hmap.put(key, new HashSet<>());
}
hmap.get(key).add(newElement);