java - 合并两个 Map

Question

这个问题已经有答案了:

Merge Map<String, List<String> Java 8 Stream (6 个回答)

已关闭 6 年前。

我有两个 HashMap，其中键是 String，值是 ArrayList。我需要以这样的方式合并这两个 map ，如果 -

1. 如果键尚不存在，则应使用给定的值映射键。
2. 如果键已经存在，我希望现有值与新值合并，即，existingList.addAll(newList)。

初始代码有点像--

对于 map 的所有键检查是否

 if (map1.containsKey("key")){
 List l = map1.get ("key");
 l.addAll(map2.get("key"));
 }
 else{
 map1.put(map2);
 }

我不关心这里的重复元素。有没有一种干净的方法可以做到这一点？我已经阅读了 Java 8 中引入的合并方法。但是由于这里的值是 List 类型，我正在寻找一种更高效、更干净的方法。

Answer 1

希望这段代码有帮助。您应该迭代其中一个列表并添加或更新另一个列表。

public static HashMap<String, ArrayList<Object>> merge(HashMap<String, ArrayList<Object>> list_1, HashMap<String, ArrayList<Object>> list_2) {
    //Iterate second hash map
    Iterator it = list_2.entrySet().iterator();
    while (it.hasNext()) {
        Map.Entry pair = (Map.Entry) it.next();
        String key = (String) pair.getKey();
        ArrayList<Object> value = (ArrayList<Object>) pair.getValue();
        if(list_1.containsKey(key)){
            //UPDATE
            list_1.replace(key, value); //Replaces the entry for the specified key only if it is currently mapped to some value.
        } else {
            //ADD
            list_1.put(key, value);
        }
    }
    return list_1;
}