首页 > 软件编程 > java > Java求两集合的交集、并集、差集

Java求两集合的交集、并集、差集实例

2024-08-02 11:15:14 作者：猫饭_ACE

这篇文章主要介绍了Java求两集合的交集、并集、差集实例,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

一、内置函数实现

removeAll方法：从list中删除指定集合中包含的所有元素。
retainAll方法：从list中删除指定集合中不包含的所有元素。
addAll方法：用来向Set集合添加另一个集合对象所包含的所有内容。

public class CollectionTest {
 
    public static void main(String[] args) {
        List<String> list1 = new ArrayList<>();
        List<String> list2 = new ArrayList<>();
 
        list1.add("a");
        list1.add("b");
        list1.add("c");
 
        list2.add("b");
        list2.add("e");
        list2.add("d");
 
        //求交集
        list1.retainAll(list2);
        System.out.println(list1.toString()); // b
 
        //求并集
        list1.removeAll(list2);
        list1.addAll(list2);
        System.out.println(list1.toString());  // a c b e d
 
        //求差集(list1有而list2没有)
        list1.removeAll(list2);   // a c
 
    }
}

二、for循环遍历实现求两个集合的交集

利用集合的contains方法，来对比第二个集合中是否存在相同的元素的方法，同时将交集结果返回。时间复杂度是O(n)。

contains方法：用于判断list集合是否包含某个元素。

public List<Object> intersectionForList_1(List<Object> arr1, List<Object> arr2) {
    long startTime = System.currentTimeMillis();
    List<Object> result = new ArrayList<>();
    for (Object arr : arr1) {
        if (arr2.contains(arr)) {
            result.add(arr);
         }
    }
    long endTime = System.currentTimeMillis();
    log.info("intersectionForList_1：" + (endTime - startTime));
    return result;
}

三、使用Java 8 Stream API 求两个集合的交集

首先把两个集合转换成Stream，然后使用filter()方法筛选出交集元素，最后把结果转换成集合。时间复杂度是O(n)。

.stream().filter()方法：来筛选出满足条件的对象，最终返回的筛选过后的列表。

Collectors.toList()和Collectors.toSet()方法都是将map后的stream转换为一个列表对象，toSet()会去重重复的对象

Set<Integer> set1 = new HashSet<>();
set1.add(1);
set1.add(2);
set1.add(3);
set1.add(4);

Set<Integer> set2 = new HashSet<>();
set2.add(3);
set2.add(4);
set2.add(5);
Set<Integer> intersection = set1.stream().filter(set2::contains).collect(Collectors.toSet());
System.out.println(intersection); // 输出 [3, 4]

四、使用迭代器求两个集合的交集

public List<Object> intersectionForList_2(List<Object> arr1, List<Object> arr2) {

      List<Object> resultList = new ArrayList<>();
 
      List<Object> maxList;
      List<Object> minList;
      if (arr1.size()>arr2.size()){
          maxList = arr1;
          minList = arr2;
      }else {
          maxList = arr2;
          minList = arr1;
      }
 
      Iterator<Object> iterator = maxList.iterator();
      while (iterator.hasNext()){
          Object next = iterator.next();
          if (minList.contains(next)){
              resultList.add(next);
          }
      }
 

      return resultList;
  }

五、使用map 求两个集合的交集

首先将list集合中的元素依次存入一个map中去，然后再以map的get方法来判断是否存在这样的元素。

这种方法的效率最高，在10万条数据的测试下，这种方法耗时仅仅在十几毫秒，而其他方法在将近一百多毫秒。

在百万条数据进行对比处理的时候，使用map特性的方法时间消耗大概在40毫秒左右，但是其他方法要将近10秒钟，

所以在对于大量数据的处理过程中，还是非常建议使用这种方法的。

public List<Object> intersectionForList_3(List<Object> arr1, List<Object> arr2) {

    List<Object> resultList = new ArrayList<>();
    Map<String,Object> map = new HashMap<>();
    arr1.forEach(a1->{
         map.put(a1+"",a1);
    });
 
    arr2.forEach(a2->{
        Object obj = map.get(a2 + "");
        if (obj!=null){
            resultList.add(obj);
        }
    });

    return resultList;
}

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。