[Java] Class TreeMap

Collection Framework

데이터 군을 저장하는 클래스들을 표준화한 설계

컬렉션 Collection :  다수의 데이터. 데이터 그룹

프레임웍 Framework :  표준화된 프로그래밍 방식.

Class TreeMap

이진 검색 트리의 형태로 키와 값의 쌍으로 이루어진 데이터를 저장한다.

검색에 관한 대부분의 경우에서는 HashMap,

범위 검색이나 정렬이 필요한 경우에는 TreeMap을 사용하는 것이 좋다.

Public constructors

Public methods

Example0

import java.util.NavigableMap;
import java.util.NavigableSet;
import java.util.TreeMap;

public class Test {
    public static void main(String[] args) {

        TreeMap<Integer, String> tm = new TreeMap<>();

        // populating tree map
        tm.put(6, "six");
        tm.put(2, "two");
        tm.put(4, "four");
        tm.put(3, "three");
        tm.put(1, "one");
        tm.put(5, "five");
        NavigableSet ns = tm.descendingKeySet();
        NavigableMap nm = tm.descendingMap();

        System.out.println(ns); // [6, 5, 4, 3, 2, 1]
        System.out.println(nm); // {6=six, 5=five, 4=four, 3=three, 2=two, 1=one}
    }
}

Example1

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Map.Entry;
import java.util.TreeMap;

public class Main {
    @SuppressWarnings("unchecked")
    public static void main(String[] args) {

        String[] data = { "A", "K", "B", "A", "C", "K", "B", "B", "D", "K", "C" };
        TreeMap<String, Integer> tm = new TreeMap<>();

        for (int i = 0; i < data.length; i++) {
            if (tm.containsKey(data[i])) {
                Integer value = tm.get(data[i]);
                tm.put(data[i], value + 1);
            } else {
                tm.put(data[i], 1);
            }
        }

        System.out.println("-- 기본(키에 대한) 정렬 --");
        for (Entry<String, Integer> map : tm.entrySet()) {
            int value = (Integer) map.getValue();
            System.out.println(map.getKey() + " : " + print('#', value) + " " + value);
        }

        List<Object> list = new ArrayList<>(tm.entrySet());
        Collections.sort(list, new Test());

        Iterator<Object> it = list.iterator();

        System.out.println("-- 값에 대한 내림차순 정렬 --");
        while (it.hasNext()) {
            Entry<String, Integer> entry = (Entry<String, Integer>) it.next();
            int value = (Integer) entry.getValue();
            System.out.println(entry.getKey() + " : " + print('#', value) + " " + value);
        }
    }

    public static String print(char ch, int value) {
        char[] bar = new char[value];

        for (int i = 0; i < bar.length; i++) {
            bar[i] = ch;
        }

        return new String(bar);
    }
}

import java.util.Comparator;
import java.util.Map.Entry;

public class Test implements Comparator<Object> {

    @Override
    public int compare(Object o1, Object o2) {
        if (o1 instanceof Entry && o2 instanceof Entry) {
            Entry e1 = (Entry) o1;
            Entry e2 = (Entry) o2;

            int v1 = (Integer) e1.getValue();
            int v2 = (Integer) e2.getValue();

            return v2 - v1;
        }
        return -1;
    }
}

-- 기본(키에 대한) 정렬 --
A : ## 2
B : ### 3
C : ## 2
D : # 1
K : ### 3
-- 값에 대한 내림차순 정렬 --
B : ### 3
K : ### 3
A : ## 2
C : ## 2
D : # 1

TreeMap API

참고 서적: 자바의 정석 3판 2