JAVA红黑树TreeMap/TreeSet详解

Java中的红黑树是一种自平衡的二叉查找树，它通过特定的旋转和重新着色操作来确保树的高度大致保持在对数级别，从而在平均和最坏情况下都能保证查找、插入、删除等操作的时间复杂度为O(log n)。红黑树通过遵循一系列的性质来维持其平衡：

红黑树的性质

1. 节点是红色或黑色：
   • 每个节点要么是红色，要么是黑色。
2. 根节点是黑色：
   • 这有助于确保从根到叶子的最长路径不会比最短路径长两倍以上。
3. 所有叶子（NIL节点，空节点）都是黑色：
   • 这里的叶子指的是树尾端的空（NIL）节点。
4. 如果一个节点是红色的，则它的两个子节点都是黑色的（也就是从每个叶子到根的所有路径上不能有两个连续的红色节点）：
   • 这有助于确保树大致上是平衡的。
5. 从任一节点到其每个叶子的所有简单路径都包含相同数目的黑色节点：
   • 这被称为黑高（black-height）性质，保证了树的高度大致是对数级别的。

使用条件

1. 需要保持元素有序：当你需要存储的元素集合需要保持有序时，红黑树是一个很好的选择。它可以确保在插入、删除和查找操作中，元素始终按照某种顺序（通常是键的自然顺序或自定义的比较器顺序）排列。

2. 需要高效的查找、插入和删除操作：红黑树通过保持树的平衡（即确保树的高度大致是对数级别的），来保证这些操作的高效性。在最坏情况下，这些操作的时间复杂度都是O(log n)，其中n是树中元素的数量。

3. 可以接受一定的额外开销：虽然红黑树提供了优秀的性能保证，但它也引入了一些额外的开销，比如插入和删除时可能需要的旋转和重新着色操作。这些操作会增加一些额外的计算成本。因此，在性能要求极高且对额外开销非常敏感的场景下，可能需要考虑其他数据结构。

使用场景

1. 数据库和索引：在数据库和文件系统中，红黑树常被用作索引结构，以加速数据的查找、插入和删除操作。例如，B+树（一种基于红黑树思想的变体）在数据库索引中非常常见。

2. 关联数组和映射：像Java中的TreeMap和TreeSet这样的集合类，内部就是使用红黑树来实现的。它们提供了按键排序的映射和集合，适用于需要保持元素有序的场景。

3. 实现优先队列：红黑树也可以用来实现优先队列（一种特殊的队列，其中每个元素都有一个优先级，元素的出队顺序按照优先级进行）。通过适当地调整节点的颜色，可以在红黑树上实现最小堆或最大堆的功能。

4. 图论算法中的数据结构：在某些图论算法中，如Dijkstra算法（用于计算图中单源最短路径）的实现中，可以使用红黑树来维护一个按距离排序的节点集合，以便高效地选择下一个要处理的节点。

5. 其他需要自平衡二叉查找树的场景：除了上述场景外，任何需要自平衡二叉查找树的场景都可以考虑使用红黑树。例如，在实现某些类型的平衡树数据结构时，红黑树可以作为一个基础选项。

 

红黑树的操作

插入

• 插入新节点：首先像普通二叉查找树一样插入新节点，并默认将其着色为红色。
• 调整：如果新节点的插入违反了红黑树的性质，则通过一系列的旋转（左旋、右旋）和重新着色操作来恢复性质。

删除

• 删除节点：首先像普通二叉查找树一样删除节点。
• 修复：删除节点后，可能需要一系列复杂的旋转和重新着色操作来恢复红黑树的性质。

Java中的红黑树实现

在Java中，TreeMap 和 TreeSet 都是通过红黑树实现的。这意味着这些集合在保持元素有序的同时，也提供了高效的查找、插入和删除操作。

示例

例1：TreeMap

import java.util.TreeMap;  
  
public class RedBlackTreeExample {  
    public static void main(String[] args) {  
        TreeMap<Integer, String> map = new TreeMap<>();  
        map.put(1, "One");  
        map.put(3, "Three");  
        map.put(2, "Two");  
  
        // TreeMap会按照键的自然顺序进行排序  
        System.out.println(map); // 输出: {1=One, 2=Two, 3=Three}  
  
        // TreeMap的键集合和值集合都是按照排序后的顺序排列的  
        System.out.println(map.keySet()); // 输出: [1, 2, 3]  
        System.out.println(map.values()); // 输出: [One, Two, Three]  
    }  
}

例2：TreeSet

import java.util.TreeSet;  
  
public class TreeSetExample {  
    public static void main(String[] args) {  
        // 创建一个 TreeSet 实例  
        TreeSet<Integer> numbers = new TreeSet<>();  
  
        // 向 TreeSet 中添加元素  
        numbers.add(3);  
        numbers.add(1);  
        numbers.add(4);  
        numbers.add(1); // 尝试添加重复元素，不会成功  
        numbers.add(2);  
  
        // 遍历并打印 TreeSet 中的元素  
        // 由于 TreeSet 是基于红黑树实现的，所以它会按照元素的自然顺序（或指定的 Comparator 顺序）进行排序  
        for (Integer number : numbers) {  
            System.out.println(number);  
        }  
  
        // 输出结果将会是排序后的元素列表：1, 2, 3, 4  
  
        // 如果你想要按照自定义的顺序对 TreeSet 中的元素进行排序，  
        // 可以在创建 TreeSet 时提供一个 Comparator 实例  
        TreeSet<String> names = new TreeSet<>((a, b) -> b.compareTo(a)); // 使用自定义比较器，按字符串的逆序排序  
  
        names.add("Alice");  
        names.add("Bob");  
        names.add("Charlie");  
  
        // 遍历并打印 names TreeSet 中的元素  
        for (String name : names) {  
            System.out.println(name);  
        }  
  
        // 输出结果将会是自定义排序后的元素列表：Charlie, Bob, Alice  
    }  
}