Java中保留java.util.Set集合前半部分元素的最优方法

首先要明确：Java标准库中的java.util.Set大部分实现（比如HashSet）是无序的，"前半部分"没有明确的定义。只有有序Set（如LinkedHashSet保留插入顺序、TreeSet按自然排序顺序）的迭代顺序是确定的，以下方案基于迭代顺序取前30000个元素。

方案1：创建新Set存储前N个元素（推荐，不修改原Set） #

这种方式不影响原集合，适合需要保留原数据的场景，代码简洁且性能稳定。

用Stream API实现（简洁直观） #

import java.util.Set;
import java.util.stream.Collectors;
import java.util.LinkedHashSet;

// 假设你的原Set是originalSet，元素类型根据实际调整
Set<String> originalSet = ...;
int keepCount = 30000;

// 保留迭代顺序的话用LinkedHashSet，不需要顺序直接用Collectors.toSet()
Set<String> retainedSet = originalSet.stream()
                                    .limit(keepCount)
                                    .collect(Collectors.toCollection(LinkedHashSet::new));

用迭代器实现（性能略优，避免Stream的额外开销） #

如果追求极致性能，手动迭代添加更直接：

import java.util.Set;
import java.util.Iterator;
import java.util.LinkedHashSet;

Set<String> originalSet = ...;
int keepCount = 30000;
Set<String> retainedSet = new LinkedHashSet<>(keepCount); // 初始化容量避免扩容
Iterator<String> iter = originalSet.iterator();
int count = 0;

while (iter.hasNext() && count < keepCount) {
    retainedSet.add(iter.next());
    count++;
}

方案2：原地修改原Set（直接删除后半部分） #

如果不需要保留原集合，且原Set支持修改（非不可变集合），可以直接删除迭代到第30000个之后的元素：

import java.util.Set;
import java.util.Iterator;

Set<String> originalSet = ...;
int keepCount = 30000;
Iterator<String> iter = originalSet.iterator();
int count = 0;

while (iter.hasNext()) {
    iter.next();
    // 计数超过保留数量时删除当前元素
    if (count >= keepCount) {
        iter.remove();
    }
    count++;
}

注意：必须用迭代器的remove()方法，否则会抛出ConcurrentModificationException。如果原Set是不可变的（比如Collections.unmodifiableSet返回的实例），不能用此方法，必须创建新Set。

关键注意事项 #

• 若使用HashSet，迭代顺序是不确定的，每次运行可能得到不同的"前半部分"元素；如果需要稳定顺序，建议先将HashSet转为LinkedHashSet再处理。
• TreeSet的迭代顺序是元素的自然排序（或自定义比较器顺序），因此取到的是排序后的前30000个元素。
• 两种方案的时间复杂度都是O(n)，空间复杂度：方案1是O(k)（k为保留元素数），方案2是O(1)（原地修改）。

内容的提问来源于stack exchange，提问作者user4202236