深入解析TreeSet与TreeMap：Java集合框架中的红黑树实现

深入解析TreeSet与TreeMap：Java集合框架中的红黑树实现

JCFInternals 项目地址: https://gitcode.com/gh_mirrors/jc/JCFInternals

概述

在Java集合框架中，TreeSet和TreeMap是两个基于红黑树实现的重要容器类。本文将深入分析它们的实现原理，帮助开发者更好地理解和使用这两个数据结构。

设计关系

TreeSet实际上是基于TreeMap实现的适配器模式典型应用。TreeSet内部持有一个TreeMap实例，所有操作都委托给这个TreeMap完成。这种设计体现了Java集合框架的巧妙复用思想。

// TreeSet内部实现片段
public class TreeSet<E> extends AbstractSet<E> {
    private transient NavigableMap<E,Object> m;
    private static final Object PRESENT = new Object();
    
    public TreeSet() {
        this.m = new TreeMap<E,Object>();
    }
    
    public boolean add(E e) {
        return m.put(e, PRESENT)==null;
    }
}

核心数据结构：红黑树

TreeMap的核心是红黑树（Red-Black Tree），这是一种自平衡的二叉查找树，具有以下特性：

1. 节点着色：每个节点非红即黑
2. 根节点：必须为黑色
3. 红色限制：红色节点的子节点必须为黑色（不能有连续红色节点）
4. 黑高一致：从任一节点到其每个叶子节点的路径包含相同数量的黑色节点

这些约束保证了红黑树的关键优势：最坏情况下基本操作（查找、插入、删除）的时间复杂度都是O(log n)。

核心操作解析

旋转操作

红黑树通过旋转操作维持平衡，包括左旋和右旋两种基本操作：

左旋过程：

1. 将x的右子树y提升为新的子树根
2. 将y的左子树变为x的右子树
3. 更新相关父节点引用

private void rotateLeft(Entry<K,V> p) {
    Entry<K,V> r = p.right;
    p.right = r.left;
    if (r.left != null) r.left.parent = p;
    r.parent = p.parent;
    // ...更新父节点引用...
    r.left = p;
    p.parent = r;
}

右旋过程与左旋对称，方向相反。

查找操作

get()方法基于二叉搜索树的查找算法：

final Entry<K,V> getEntry(Object key) {
    Comparable<? super K> k = (Comparable<? super K>) key;
    Entry<K,V> p = root;
    while (p != null) {
        int cmp = k.compareTo(p.key);
        if (cmp < 0) p = p.left;
        else if (cmp > 0) p = p.right;
        else return p;
    }
    return null;
}

插入操作

put()方法实现包含三个主要步骤：

1. 按照二叉搜索树规则找到插入位置
2. 创建新节点并插入
3. 通过fixAfterInsertion()调整树结构

插入后的调整是红黑树最复杂的部分，需要考虑多种情况并通过变色和旋转恢复平衡。

删除操作

remove()操作同样复杂，主要步骤：

1. 定位要删除的节点
2. 分三种情况处理：
   • 无子节点：直接删除
   • 一个子节点：用子节点替代
   • 两个子节点：用后继节点替代后删除
3. 通过fixAfterDeletion()调整树结构

性能特点

1. 时间复杂度：

   • 查找、插入、删除：O(log n)
   • 遍历：O(n)

2. 空间复杂度：O(n)

3. 有序性：元素按照键的自然顺序或Comparator顺序排列

4. 线程安全：非同步，多线程环境下需要外部同步

使用建议

1. 当需要有序集合时优先考虑TreeSet/TreeMap
2. 对于频繁插入删除的场景，相比HashMap有额外平衡开销
3. 自定义对象作为键时，需正确实现Comparable或提供Comparator
4. 多线程环境应使用Collections.synchronizedSortedMap包装

实现细节精要

1. 后继节点查找：删除操作中需要找到节点的后继（中序遍历的下一个节点）

static <K,V> Entry<K,V> successor(Entry<K,V> t) {
    if (t.right != null) { // 情况1：右子树存在
        Entry<K,V> p = t.right;
        while (p.left != null) p = p.left;
        return p;
    } else { // 情况2：向上查找第一个左祖先
        Entry<K,V> p = t.parent;
        Entry<K,V> ch = t;
        while (p != null && ch == p.right) {
            ch = p;
            p = p.parent;
        }
        return p;
    }
}

2. 插入调整：处理红红冲突，通过变色和旋转恢复平衡
3. 删除调整：处理黑高变化，通过复杂的条件判断和旋转操作

总结

TreeSet和TreeMap作为Java集合框架中有序容器的代表，其红黑树实现提供了良好的性能保证。理解它们的内部实现机制，有助于开发者根据具体场景做出更合理的数据结构选择，并能够更有效地使用这些工具类。红黑树的平衡策略虽然增加了实现的复杂性，但换来了稳定的性能表现，这也是它被Java集合框架采用的原因。

JCFInternals 项目地址: https://gitcode.com/gh_mirrors/jc/JCFInternals