Java inverser une carte

Question

J'ai besoin d'inverser une carte originale. lequel type est <Integer, String>, comme {1 = A, 2 = A, 3 = B....}. Je veux créer une nouvelle carte qui est String à ArrayList parce que si 1 = A, et 2 = A, que je veux avoir quelque chose comme: A = [1， 2].

Alors, comment puis-je faire cela?

Answer 1

Vous pouvez essayer ceci:

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

Alors, disons que HashMap<Integer, String> original est {1=A, 2=B, 3=C, 4=A}, alors vous obtiendrez {A=[1, 4], B=[2], C=[3]}.

EDIT: Si vous voulez une version plus générique, comme @ Mr.Polywhirl a suggéré, vous pouvez utiliser:

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

Vous pouvez facilement le faire en utilisant Java 8 API stream, ci-dessous est un exemple :

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

}