Java的Set集合，你真的会用吗？HashSet/TreeSet/LinkedHashSet

一灯架构2023-11-30 20:25

这几天拼多多的市值超过阿里了，热衷砍一刀的电商小弟竟然干翻电商老大，实在让人费解，是大家都没钱了还是啥原因？拼多多可是没有天猫、没有支付宝、没有云服务、没有钉钉、没有大文娱，就靠一个社交电商。

阿里这艘大船是否要翻不清楚，大家期待的《解读Java源码专栏》不能停止更新，在这个系列中，我将手把手带着大家剖析Java核心组件的源码，内容包含集合、线程、线程池、并发、队列等，深入了解其背后的设计思想和实现细节，轻松应对工作面试。

这是解读Java源码系列的第七篇，将跟大家一起学习Java中Set集合。

引言

当我们需要对元素去重的时候，会使用Set集合，可选的Set集合有三个，分别是HashSet、LinkedHashSet、TreeSet，这三个常用的Set集合有什么区别呢？底层实现原理是什么样？这篇文章一起来深度剖析。

共同点

这三个类都实现了Set接口，所以使用方式都是一样的，使用add()方法添加元素，使用remove()删除元素，使用contains()方法判断元素是否存在，使用iterator()方法迭代遍历元素，这三个类都可以去除重复元素。

特性

  1. HashSet是最基础的Set集合，可以去除重复元素。
  2. LinkedHashSet在HashSet功能基础上，增加了按照元素插入顺序或者访问顺序的迭代方式。
  3. TreeSet在HashSet功能基础上，可以保证按照元素大小顺序排列。

底层实现

  1. HashSet是基于HashMap实现的，使用组合的方式，并非继承。
  2. LinkedHashSet继承自HashSet，而内部则是采用组合LinkedHashMap的方式实现的。[流汗] 就是这么乱，一会儿看一下源码就明白了。
  3. TreeSet是基于TreeMap实现的，采用组合的方式，跟上面两个Set集合没关系。

下面详细看一下这三个Set集合源码的底层实现：

HashSet源码实现

类属性

java 复制代码 
public class HashSet<E>
        extends AbstractSet<E>
        implements Set<E>, Cloneable, java.io.Serializable {

    /**
     * 使用HashMap存储数据
     */
    private transient HashMap<E, Object> map;

    /**
     * value的默认值
     */
    private static final Object PRESENT = new Object();

}

可以看出HashSet实现了Set接口，内部采用HashMap存储元素，利用了HashMap的key不能重复的特性，实现元素去重。而value使用默认值，是一个空对象，没有任何作用，纯粹占坑。

初始化

HashSet常用的构造方法有两个，有参构造方法，可以指定初始容量和负载系数。

java 复制代码 
/**
 * 无参构造方法
 */
HashSet<Integer> hashSet1 = new HashSet<>();

/**
 * 有参构造方法，指定初始容量和负载系数
 */
HashSet<Integer> hashSet = new HashSet<>(16, 0.75f);

再看一下构造方式对应的源码实现：

java 复制代码 
/**
 * 无参构造方法
 */
public HashSet() {
    map = new HashMap<>();
}

/**
 * 有参构造方法，指定初始容量和负载系数
 */
public HashSet(int initialCapacity, float loadFactor) {
    map = new HashMap<>(initialCapacity, loadFactor);
}

HashSet的构造方式源码也很简单，都是利用的HashMap的构造方法实现。

常用方法源码

再看一下HashSet常用方法源码实现：

java 复制代码 
/**
 * 添加元素
 */
public boolean add(E e) {
    return map.put(e, PRESENT) == null;
}

/**
 * 删除元素
 */
public boolean remove(Object o) {
    return map.remove(o) == PRESENT;
}

/**
 * 判断是否包含元素
 */
public boolean contains(Object o) {
    return map.containsKey(o);
}

/**
 * 迭代器
 */
public Iterator<E> iterator() {
    return map.keySet().iterator();
}

HashSet方法源码也很简单，都是利用HashMap的方法实现逻辑。利用HashMap的key不能重复的特性，value使用默认值，contains()方法和iterator()方法也都是针对key进行操作。

LinkedHashSet源码实现

类属性

LinkedHashSet继承自HashSet，没有任何私有的属性。

java 复制代码 
public class LinkedHashSet<E>
        extends HashSet<E>
        implements Set<E>, Cloneable, java.io.Serializable {
}

初始化

LinkedHashSet常用的构造方法有三个，有参构造方法，可以指定初始容量和负载系数。

java 复制代码 
/**
 * 无参构造方法
 */
Set<Integer> linkedHashSet1 = new LinkedHashSet<>();

/**
 * 有参构造方法，指定初始容量
 */
Set<Integer> linkedHashSet2 = new LinkedHashSet<>(16);

/**
 * 有参构造方法，指定初始容量和负载系数
 */
Set<Integer> linkedHashSet3 = new LinkedHashSet<>(16, 0.75f);

再看一下构造方法的源码实现：

java 复制代码 
/**
 * 无参构造方法
 */
public LinkedHashSet() {
    super(16, .75f, true);
}

/**
 * 有参构造方法，指定初始容量
 */
public LinkedHashSet() {
    super(16, .75f, true);
}

/**
 * 有参构造方法，指定初始容量和负载系数
 */
public LinkedHashSet(int initialCapacity, float loadFactor) {
    super(initialCapacity, loadFactor, true);
}

LinkedHashSet的构造方法使用的是父类HashSet的构造方法，而HashSet的构造方法使用的是LinkedHashMap的构造方法，设计的就是这么乱！

java 复制代码 
public class HashSet<E>
        extends AbstractSet<E>
        implements Set<E>, Cloneable, java.io.Serializable {

    /**
     * HashSet的构造方法，底层使用的是LinkedHashMap，专门给LinkedHashSet使用
     *
     * @param initialCapacity 初始容量
     * @param loadFactor      负载系数
     * @param dummy           这个字段没啥用
     */
    HashSet(int initialCapacity, float loadFactor, boolean dummy) {
        map = new LinkedHashMap<>(initialCapacity, loadFactor);
    }

}

LinkedHashSet的其他方法也是使直接用的父类HashSet的方法，就不用看了。

LinkedHashSet额外实现了按照元素的插入顺序或者访问顺序进行迭代的功能，是使用LinkedHashMap的实现，不了解LinkedHashMap的，可以看一下上篇文章对LinkedHashMap的源码解析。

TreeSet源码实现

类属性

java 复制代码 
public class TreeSet<E> extends AbstractSet<E>
        implements NavigableSet<E>, Cloneable, java.io.Serializable {

    /**
     * 用来存储数据
     */
    private transient NavigableMap<E, Object> m;

    /**
     * value的默认值
     */
    private static final Object PRESENT = new Object();

}

TreeSet内部使用NavigableMap存储数据，而NavigableMap是TreeMap的父类，后面在初始化NavigableMap的时候，会用TreeMap进行替换。而value使用默认空对象，与HashSet类似。

初始化

TreeSet有两个构造方法，有参构造方法，可以指定排序方式，默认是升序。

java 复制代码 
/**
 * 无参构造方法
 */
TreeSet<Integer> treeSet1 = new TreeSet<>();

/**
 * 有参构造方法，传入排序方式，默认升序，这里传入倒序
 */
TreeSet<Integer> treeSet2 = new TreeSet<>(Collections.reverseOrder());

再看一下构造方法的源码实现：

java 复制代码 
TreeSet(NavigableMap<E,Object> m) {
    this.m = m;
}

/**
 * 无参构造方法
 */
public TreeSet() {
    this(new TreeMap<E, Object>());
}

/**
 * 有参构造方法，传入排序方式，默认升序，这里传入倒序
 */
public TreeSet(Comparator<? super E> comparator) {
    this(new TreeMap<>(comparator));
}

TreeSet的构造方法内部是直接使用的TreeMap的构造方法，是基于TreeMap实现的。

常用方法源码

java 复制代码 
/**
 * 添加元素
 */
public boolean add(E e) {
    return m.put(e, PRESENT) == null;
}

/**
 * 删除元素
 */
public boolean remove(Object o) {
    return m.remove(o) == PRESENT;
}

/**
 * 判断是否包含元素
 */
public boolean contains(Object o) {
    return m.containsKey(o);
}

/**
 * 迭代器
 */
public Iterator<E> iterator() {
    return m.navigableKeySet().iterator();
}

TreeSet常用方法的底层实现都是使用的TreeMap的方法逻辑，就是这么偷懒。

TreeSet可以按元素大小顺序排列的功能，也是使用TreeMap实现的，感兴趣的可以看一下上篇文章讲的TreeMap源码。由于TreeSet可以元素大小排列，所以跟其他Set集合相比，增加了一些按照元素大小范围查询的方法。

其他方法列表：

作用 方法签名
获取第一个元素 E first()
获取最后一个元素 E last()
获取大于指定键的最小键 E higher(E e)
获取小于指定键的最大元素 E lower(E e)
获取大于等于指定键的最小键 E ceiling(E e)
获取小于等于指定键的最大键 E floor(E e)
获取并删除第一个元素 E pollFirst()
获取并删除最后一个元素 E pollLast()
获取前几个元素（inclusive表示是否包含当前元素） NavigableSet headSet(E toElement, boolean inclusive)
获取后几个元素（inclusive表示是否包含当前元素） NavigableSet tailSet(E fromElement, boolean inclusive)
获取其中一段元素集合（inclusive表示是否包含当前元素） NavigableSet subSet(E fromElement, boolean fromInclusive, E toElement, boolean toInclusive)
获取其中一段元素集合（左开右开） SortedSet subSet(E fromElement, E toElement)
获取前几个元素（不包含当前元素） SortedSet headSet(E toElement)
获取后几个元素（不包含当前元素） SortedSet tailSet(E fromElement)

总结

HashSet、LinkedHashSet、TreeSet，这三个常用的Set集合的共同点是都实现了Set接口，所以使用方式都是一样的，使用add()方法添加元素，使用remove()删除元素，使用contains()方法判断元素是否存在，使用iterator()方法迭代遍历元素，这三个类都可以去除重复元素。

不同点是：

HashSet的关键特性：

  1. 是最基础的Set集合，可以去除重复元素。
  2. HashSet是基于HashMap实现的，使用组合的方式，并非继承。
  3. 利用了HashMap的key不重复的特性，而value是一个默认空对象，其他方法也都是使用HashMap实现。

LinkedHashSet的关键特性：

  1. LinkedHashSet继承自HashSet，而内部则是采用组合LinkedHashMap的方式实现的。
  2. LinkedHashSet在HashSet功能基础上，增加了按照元素插入顺序或者访问顺序的迭代方式，代价是额外增加一倍的存储空间。
  3. 方法内部都是使用LinkedHashMap实现的。

Tr**eeSet**的关键特性：

  1. TreeSet是基于TreeMap实现的，也是采用组合的方式。
  2. TreeSet在HashSet功能基础上，可以保证按照元素大小顺序排列，代价是查询、插入、删除接口的时间复杂度从O(1)退化到O(log n)。
  3. 方法内部都是使用TreeMap实现的。
相关推荐
Estar.Lee几秒前
查手机号归属地免费API接口教程
android·网络·后端·网络协议·tcp/ip·oneapi
弗拉唐1 小时前
springBoot，mp，ssm整合案例
java·spring boot·mybatis
oi771 小时前
使用itextpdf进行pdf模版填充中文文本时部分字不显示问题
java·服务器
2401_857610032 小时前
SpringBoot社团管理：安全与维护
spring boot·后端·安全
少说多做3432 小时前
Android 不同情况下使用 runOnUiThread
android·java
知兀2 小时前
Java的方法、基本和引用数据类型
java·笔记·黑马程序员
蓝黑20202 小时前
IntelliJ IDEA常用快捷键
java·ide·intellij-idea
Ysjt | 深2 小时前
C++多线程编程入门教程（优质版）
java·开发语言·jvm·c++
凌冰_2 小时前
IDEA2023 SpringBoot整合MyBatis（三）
spring boot·后端·mybatis
热门推荐
01ICLR24大模型提示(1/11) | BadChain：大型语言模型的后门思维链提示02PyTorch机器学习实现液态神经网络03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04【音视频 | opus】opus编解码库(opus-1.4)详细介绍以及使用——附带解码示例代码05Coze扣子平台完整体验和实践（附国内和国际版对比）06(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明07IT专业入门，高考假期预习指南08玄机平台应急响应—webshell查杀09Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO10〔AI 绘画〕Stable Diffusion 之 解决绘制多人或面部很小的人物时面部崩坏问题 篇