[Java] Map을 값(Value) 기준으로 정렬하는 방법

1. Map을 값 기준으로 정렬해야 하는 이유

Java에서 Map<Key, Value>는 기본적으로 키(Key) 기준으로 정렬됩니다. 그러나 값(Value)을 기준으로 정렬해야 하는 경우가 있습니다. 예를 들어, 빈도수에 따라 항목을 정렬하거나 특정 기준에 따라 데이터를 표시할 때 유용합니다.

---

2. Java에서 Map을 값 기준으로 정렬하는 방법

2.1 Java 8 Stream API를 활용한 정렬

Java 8부터는 Stream API를 사용해 Map을 간단히 정렬할 수 있습니다.

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

Map<String, Integer> sortedMap = map.entrySet().stream()
    .sorted(Map.Entry.comparingByValue())
    .collect(Collectors.toMap(
        Map.Entry::getKey,
        Map.Entry::getValue,
        (e1, e2) -> e1,
        LinkedHashMap::new
    ));

System.out.println(sortedMap);

결과

{B=1, C=2, A=3}

---

2.2 Java 8 이전의 방식: Comparator와 List 사용

Java 8 이전 버전에서는 다음과 같이 정렬할 수 있습니다.

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

List<Map.Entry<String, Integer>> list = new ArrayList<>(map.entrySet());
list.sort((o1, o2) -> o1.getValue().compareTo(o2.getValue()));

Map<String, Integer> sortedMap = new LinkedHashMap<>();
for (Map.Entry<String, Integer> entry : list) {
    sortedMap.put(entry.getKey(), entry.getValue());
}

System.out.println(sortedMap);

결과

{B=1, C=2, A=3}

---

2.3 사용자 정의 Comparator를 사용한 TreeMap 정렬

TreeMap에 사용자 정의 Comparator를 적용하면 값 기준으로 정렬할 수 있습니다.

class ValueComparator implements Comparator<String> {
    Map<String, Integer> map;

    public ValueComparator(Map<String, Integer> map) {
        this.map = map;
    }

    @Override
    public int compare(String o1, String o2) {
        return map.get(o1).compareTo(map.get(o2));
    }
}

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

TreeMap<String, Integer> sortedMap = new TreeMap<>(new ValueComparator(map));
sortedMap.putAll(map);

System.out.println(sortedMap);

결과

{B=1, C=2, A=3}

---

3. 주의 사항 및 팁

3.1 값 중복 처리

값이 중복되는 경우 TreeMap은 일부 키를 무시할 수 있습니다. 이를 방지하려면 중복 값 처리 로직을 추가해야 합니다.

3.2 성능 고려

큰 데이터셋을 처리할 때는 Stream API를 사용해 성능을 최적화하는 것이 유리합니다.

3.3 정렬 결과 보장

LinkedHashMap을 사용하면 삽입된 순서를 유지할 수 있습니다.

---

결론

Java에서 Map을 값 기준으로 정렬하려면 Stream API, Comparator, 또는 TreeMap을 적절히 활용하세요. 프로젝트에 맞는 방식을 선택하여 효율적으로 데이터를 정렬하고 관리할 수 있습니다.