Java反转地图

Question

我需要反转原始地图。哪个类型是<Integer, String>，比如{1 = A, 2 = A, 3 = B....}。我想创建一个从String 到ArrayList 的新地图，因为如果1 = A 和2 = A，我想要这样的东西：A = [1， 2]。

那我该怎么做呢？

Answer 1

您可以使用 Java 8 的 stream API 轻松做到这一点，下面是一个示例：

public static void main(String[] args) throws FileNotFoundException {

    Map<Integer, String> map = new HashMap<>();
    map.put(1, "A");
    map.put(2, "A");
    map.put(3, "B");

    Map<String, List<Integer>> invertedMap = map.entrySet()
    .stream()
    .collect(Collectors.groupingBy(Entry::getValue, 
            Collectors.mapping(Entry::getKey, Collectors.toList())));

    System.out.println(invertedMap);

}

Answer 2

你可以试试这个：

HashMap<Integer, String> original = new HashMap<>();
HashMap<String, ArrayList<Integer>> inverted = new HashMap<>();

original.put(1, "A");
original.put(2, "B");
original.put(3, "C");
original.put(4, "A");

for (Integer key: original.keySet()) {
    String newKey = original.get(key);

    inverted.computeIfAbsent(newKey, k -> new ArrayList<>());
    inverted.get(newKey).add(key);

}
System.out.println(original);
System.out.println(inverted);

所以，假设HashMap<Integer, String> original 是{1=A, 2=B, 3=C, 4=A}，那么你将得到{A=[1, 4], B=[2], C=[3]}。

编辑：如果您想要更通用的版本，正如 @Mr.Polywhirl 所建议的，您可以使用：

public static final <T, U> Map<U, List<T>> invertMap(Map<T, U> map) {
    HashMap<U, List<T>> invertedMap = new HashMap<>();

    for (T key : map.keySet()) {
        U newKey = map.get(key);

        invertedMap.computeIfAbsent(newKey, k -> new ArrayList<>());
        invertedMap.get(newKey).add(key);

    }

    return invertedMap;
}