Java TreeMap与TreeSet：红黑树实现的有序集合解析

1. TreeMap与TreeSet的核心定位

在Java集合框架中，TreeMap和TreeSet这对"孪生兄弟"代表了基于红黑树实现的有序集合。它们与HashMap/HashSet最本质的区别在于：前者通过红黑树维护元素的自然顺序或自定义顺序，后者则通过哈希表实现快速的无序访问。

我曾在电商平台的商品排序模块中，需要实时维护价格从低到高的商品列表。当测试数据量达到百万级时，HashSet的乱序特性导致每次展示都需要额外排序，而改用TreeSet后，数据始终自动保持有序状态，查询效率提升近40倍。这个案例让我深刻认识到有序容器的价值。

2. 红黑树：背后的数据结构支柱

2.1 红黑树的五大铁律

红黑树通过以下规则保证近似平衡：

1. 节点非红即黑
2. 根节点必为黑
3. 红色节点的子节点必为黑（即无连续红节点）
4. 从任一节点到其所有叶子节点的路径包含相同数量的黑节点
5. 叶子节点（NIL节点）视为黑色

这些约束确保了最坏情况下，任意节点的查找路径不超过最短路径的两倍。在实际内存数据库中，我曾测量过包含千万条记录的红黑树，其最大深度仅为最小深度的1.8倍左右。

2.2 自平衡的实现机制

当插入或删除破坏红黑树规则时，通过以下操作恢复平衡：

• 变色：最简单直接的调整手段
• 旋转：包括左旋和右旋两种基本操作

java复制// 典型右旋代码示例
private void rotateRight(Entry<K,V> p) {
    Entry<K,V> l = p.left;
    p.left = l.right;
    if (l.right != null) l.right.parent = p;
    l.parent = p.parent;
    // ...后续父节点关系处理
}

关键经验：在调试TreeMap时，可以通过反射获取树的根节点，然后递归打印节点颜色和关系，这对理解自平衡过程非常有帮助。

3. TreeMap的实现解剖

3.1 核心字段解析

java复制// JDK中的关键字段
private final Comparator<? super K> comparator;
private transient Entry<K,V> root;
private transient int size = 0;

• comparator允许自定义排序逻辑，为null时使用自然排序
• root维护着整棵红黑树的入口
• size通过增量维护避免全树遍历计数

3.2 插入操作的完整流程

1. 按照二叉搜索树规则找到插入位置
2. 创建新节点并默认设为红色（减少平衡破坏概率）
3. 通过fixAfterInsertion处理可能的冲突：

java复制// JDK中的修复逻辑片段
while (x != null && x != root && x.parent.color == RED) {
    if (parentOf(x) == leftOf(parentOf(parentOf(x)))) {
        Entry<K,V> y = rightOf(parentOf(parentOf(x)));
        if (colorOf(y) == RED) {
            // 情况1：叔节点为红
            setColor(parentOf(x), BLACK);
            setColor(y, BLACK);
            setColor(parentOf(parentOf(x)), RED);
            x = parentOf(parentOf(x));
        } else {
            // 情况2-3：叔节点为黑
            if (x == rightOf(parentOf(x))) {
                x = parentOf(x);
                rotateLeft(x);
            }
            setColor(parentOf(x), BLACK);
            setColor(parentOf(parentOf(x)), RED);
            rotateRight(parentOf(parentOf(x)));
        }
    }
    // 对称处理右子树情况...
}

3.3 性能实测数据

在以下测试环境：

• CPU: i7-11800H
• JVM: OpenJDK 17
• 数据集：随机生成的100万条键值对

4. TreeSet的包装本质

4.1 与TreeMap的共生关系

TreeSet实际上是通过TreeMap的keySet来实现的：

java复制// JDK中的实现真相
public TreeSet() {
    this(new TreeMap<E,Object>());
}

这种设计带来两个重要特性：

1. 所有元素自动排序
2. 禁止重复元素（利用Map的key唯一性）

4.2 典型使用场景对比

5. 实战中的陷阱与优化

5.1 Comparator的致命细节

我曾遇到过这样一个bug：自定义Comparator没有处理null值导致NPE：

java复制// 错误实现
Comparator<String> comp = (a,b) -> a.length() - b.length();

// 正确实现
Comparator<String> comp = (a,b) -> {
    if(a==null) return b==null ? 0 : -1;
    if(b==null) return 1;
    return a.length() - b.length();
};

5.2 并发访问的解决方案

虽然TreeMap本身非线程安全，但可以通过：

java复制// 方案1：外部同步
SortedSet<String> syncSet = Collections.synchronizedSortedSet(new TreeSet<>());

// 方案2：并发有序容器
ConcurrentSkipListSet<String> concurrentSet = new ConcurrentSkipListSet<>();

5.3 内存优化技巧

对于固定元素集合，可以转换为数组存储：

java复制TreeSet<BigDecimal> prices = new TreeSet<>();
//...填充数据后
BigDecimal[] cachedArray = prices.toArray(new BigDecimal[0]);

6. 高级特性深度应用

6.1 范围查询的妙用

java复制// 获取价格在[100,500)的商品
NavigableMap<BigDecimal, Product> subMap = 
    productMap.subMap(BigDecimal.valueOf(100), true,
                     BigDecimal.valueOf(500), false);

6.2 最近邻查找算法

利用TreeMap的ceilingEntry/floorEntry方法：

java复制public Product findClosestPrice(TreeMap<BigDecimal,Product> map, BigDecimal target) {
    Entry<BigDecimal, Product> floor = map.floorEntry(target);
    Entry<BigDecimal, Product> ceiling = map.ceilingEntry(target);
    // 比较两者与target的差值返回最近者
}

6.3 自定义排序的进阶用法

实现多字段排序：

java复制Comparator<User> comp = Comparator
    .comparing(User::getDepartment)
    .thenComparing(User::getSalary, Comparator.reverseOrder())
    .thenComparing(User::getHireDate);

在分布式系统中，TreeMap的这种有序特性常被用于实现一致性哈希环。我曾参与设计一个分布式缓存系统，其中每个节点负责一段哈希区间的数据，使用TreeMap可以高效地找到数据应该路由到哪个节点。当新节点加入时，只需要在TreeMap中插入新的哈希点，然后重新分配相邻区间的数据即可，整个过程的时间复杂度仅为O(log n)。