Java翻轉地圖

Question

我需要翻轉原始地圖。其類型爲<Integer, String>，如{1 = A, 2 = A, 3 = B....}。我想創建一個新的地圖，這是String到ArrayList，因爲如果1 = A和2 = A，比我想要這樣的東西：A = [1， 2]。

那麼我該怎麼做呢？

Answer 1

你可以試試這個：

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; 
} 

Answer 2

你可以輕鬆地使用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); 

}