자바가지도를 뒤집습니다.

Question

원본지도를 뒤집을 필요가 있습니다. 어떤 유형이 <Integer, String>입니다 (예 : {1 = A, 2 = A, 3 = B....}). String ~ ArrayList 인 새로운지도를 만들고 싶습니다. 과 같은 것을 가지고 있기 때문에 1 = A 및 2 = A 인 경우보다 유용합니다.

어떻게하면됩니까?

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

쉽게 아래 예를 들어, 그것은 자바 (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); 

}