Що таке TreeMap в Java?

Клас TreeMap у Java представляє реалізацію інтерфейсу Map, що базується на червоно-чорному дереві. Це означає, що ключі у колекції автоматично сортуються відповідно до їх природного порядку або заданого компаратора.

TreeMap забезпечує ефективний доступ до даних, дозволяючи швидко знаходити, додавати чи видаляти елементи. Завдяки своїй структурі, він підходить для завдань, де важливий порядок елементів або необхідно виконувати операції з діапазонами.

1. Елементи впорядковані: ключі завжди зберігаються у відсортованому порядку.
2. Не допускає null як ключів: спроба вставити null викличе виняток NullPointerException.
3. Ітерація по ключах, значеннях або парах ключ-значення здійснюється у порядку сортування.
4. Підтримка методів пошуку діапазонів: можна отримувати піддерева за допомогою subMap(), headMap() та tailMap().

import java.util.TreeMap;

public class TreeMapExample {
    public static void main(String[] args) {
        // Створюємо TreeMap
        TreeMap<Integer, String> treeMap = new TreeMap<>();

        // Додаємо елементи
        treeMap.put(1, "Один");
        treeMap.put(3, "Три");
        treeMap.put(2, "Два");

        // Виводимо елементи у відсортованому порядку
        System.out.println("TreeMap: " + treeMap);

        // Доступ до значень за ключем
        System.out.println("Значення для ключа 2: " + treeMap.get(2));

        // Видалення елемента
        treeMap.remove(1);
        System.out.println("Після видалення ключа 1: " + treeMap);

        // Використання піддіапазонів
        System.out.println("Елементи менше 3: " + treeMap.headMap(3));
    }
}

TreeMap — це потужний інструмент для роботи з даними, де важливо зберегти порядок ключів. Наприклад, його можна використовувати для зберігання інформації про записи у базі даних, де ключі є ідентифікаторами записів, або для обробки часових міток у хронологічному порядку.

Проте варто враховувати, що операції з TreeMap трохи повільніші, ніж з HashMap, через необхідність підтримувати структуру дерева. Якщо порядок ключів не важливий, HashMap може бути більш ефективним вибором.

TreeMap також підтримує передавання власного компаратора, що дозволяє реалізувати кастомні правила сортування. Це може бути корисно, наприклад, для роботи з об'єктами складних типів, де необхідно враховувати специфічні поля об'єкта.