生成一个Map，其中包含两个map Map中包含的值的联合

Question

我有两个map参数：Map

我正在寻找一种方法来编写一个计算它们的并集的函数。我的意思是，对于任何X x，如果它在map1的域中，但不在map2的域中，那么它的值将是map1.get（x）。相反的情况相同。如果它在两者中，那么我想返回一个集合，它是map1.get（x）和map2.get（x）的交集。

假设我知道X是哪个类，可以使用以下代码完成此操作：

public Map<X,Set<Y>> unifyAndIntersect(Map<X,Set<Y>> map1, Map<X,Set<Y>> map2) {
    Map<X,Set<Y>> combined = new HashMap();
    combined.putAll(map1);
    for(X x : map2.keySet()){
          if(combined.contains(x)){
                Set<Y> res = Sets.newSet();
                res.addAll(map1.get(x));
                res.retainAll(map2.get(x));
                combined.put(x,res);
          }
          else{
                combined.put(x,map2.get(x));
          }
    }
}

但是，我想使这个方法通用，从某种意义上说，它适用于任何X和Y。我尝试过使用Object，但从我的类类型转换为Object时出错...

你能告诉我做这件事的正确方法吗？




Answer 1

为了声明泛型方法，需要提供泛型类型参数

别忘了钻石

您提供的代码中还有一个不一致之处：如果一个键存在于两个映射中，则会将一个新集作为值添加到结果映射中，但如果根据您的代码，该键仅包含在其中一个映射中，则会使用现有集。我最好为每个值生成一个新的集，以确保随后对结果图的值进行的修改不会影响map1和map2的状态。

public <X,Y> Map<X, Set<Y>> unifyAndIntersect(Map<X,Set<Y>> map1,
                                              Map<X,Set<Y>> map2) {
    
    Map<X, Set<Y>> combined = new HashMap<>();
    for(Map.Entry<X, Set<Y>> entry: map1.entrySet()){
        Set<Y> union = new HashSet<>(entry.getValue());
        if (map2.containsKey(entry.getKey())) {
            union.addAll(map2.get(entry.getKey()));
        }
        combined.put(entry.getKey(), union);
    }
    return combined;
}

可以使用流API实现相同的逻辑：

public <X,Y> Map<X, Set<Y>> unifyAndIntersect(Map<X,Set<Y>> map1,
                                              Map<X,Set<Y>> map2) {
    return map1.entrySet().stream()
        .map(entry -> Map.entry(entry.getKey(),
                                Stream.of(entry.getValue(), 
                                          map2.get(entry.getKey()))
                    .filter(Objects::nonNull)
                    .flatMap(Set::stream)
                    .collect(Collectors.toSet())))
        .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
}