如何从多个TreeSet中提取交集元素并合并为一个TreeSet？

嘿，要解决多个TreeSet的交集问题其实不难，Java集合框架里就有现成的方法可以用，我给你两种实用的方案，你可以根据场景选择：

方案一：用retainAll逐步求交集 #

这个思路超直接：先拿第一个TreeSet作为基础结果，然后依次和剩下的每个TreeSet做交集操作——retainAll方法会直接修改原集合，只保留两个集合共有的元素。

假设你已经把所有要处理的TreeSet放进了一个列表List<TreeSet<String>> setList，代码可以这么写：

// 先判断集合列表是否为空，避免空指针异常
if (setList == null || setList.isEmpty()) {
    return new TreeSet<>();
}

// 以第一个集合为初始交集结果（注意要新建一个集合，避免修改原集合）
TreeSet<String> intersection = new TreeSet<>(setList.get(0));

// 依次和剩下的集合求交集
for (int i = 1; i < setList.size(); i++) {
    // 如果某次交集后结果为空，直接提前终止，没必要继续遍历了
    if (intersection.isEmpty()) {
        break;
    }
    intersection.retainAll(setList.get(i));
}

// 最终intersection就是所有集合共有的元素

这种方法的优点是简单直观，代码量少，而且TreeSet的retainAll基于红黑树实现，效率也挺不错。

方案二：统计元素出现次数（适合多集合场景） #

如果你的TreeSet数量特别多，或者不想修改原集合，也可以用统计次数的方式：把所有TreeSet里的元素放进HashMap统计出现次数，最后取出那些出现次数等于TreeSet总数的元素——这些就是所有集合都共有的元素。

代码示例：

if (setList == null || setList.isEmpty()) {
    return new TreeSet<>();
}

Map<String, Integer> countMap = new HashMap<>();
int totalSetCount = setList.size();

// 遍历所有集合，统计每个元素的出现次数
for (TreeSet<String> set : setList) {
    for (String element : set) {
        countMap.put(element, countMap.getOrDefault(element, 0) + 1);
    }
}

// 筛选出出现次数等于集合总数的元素，存入TreeSet
TreeSet<String> intersection = new TreeSet<>();
for (Map.Entry<String, Integer> entry : countMap.entrySet()) {
    if (entry.getValue() == totalSetCount) {
        intersection.add(entry.getKey());
    }
}

return intersection;

给你的代码片段提个小建议 #

你提到用TreeSet[] results = new TreeSet[uniqueWords.size()]这种数组方式，其实换成List<TreeSet<String>>会更灵活安全，还能避免数组的类型转换问题。另外，不管用哪种方案，都要注意空集合的判断，不然很容易踩空指针的坑哦。

拿你举的例子来说：TreeSet1有a、b、c、d，TreeSet2有b、d、e、g，TreeSet3有b、e、d、h，用第一种方案的话，初始intersection是{a,b,c,d}，和TreeSet2做retainAll后变成{b,d}，再和TreeSet3做retainAll还是{b,d}，正好得到你想要的结果~

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