快捷導(dǎo)航

Java求兩集合的交集、并集、差集實例

更新時間：2024年08月02日 11:15:43 作者：貓飯_ACE

這篇文章主要介紹了Java求兩集合的交集、并集、差集實例,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

一、內(nèi)置函數(shù)實現(xiàn)

removeAll方法：從list中刪除指定集合中包含的所有元素。
retainAll方法：從list中刪除指定集合中不包含的所有元素。
addAll方法：用來向Set集合添加另一個集合對象所包含的所有內(nèi)容。

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循環(huán)遍歷實現(xiàn) 求兩個集合的交集

利用集合的contains方法，來對比第二個集合中是否存在相同的元素的方法，同時將交集結(jié)果返回。時間復(fù)雜度是O(n)。

contains方法：用于判斷l(xiāng)ist集合是否包含某個元素。

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 求兩個集合的交集

首先把兩個集合轉(zhuǎn)換成Stream，然后使用filter()方法篩選出交集元素，最后把結(jié)果轉(zhuǎn)換成集合。時間復(fù)雜度是O(n)。

.stream().filter()方法：來篩選出滿足條件的對象，最終返回的篩選過后的列表。

Collectors.toList()和Collectors.toSet()方法都是將map后的stream轉(zhuǎn)換為一個列表對象，toSet()會去重重復(fù)的對象

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萬條數(shù)據(jù)的測試下，這種方法耗時僅僅在十幾毫秒，而其他方法在將近一百多毫秒。

在百萬條數(shù)據(jù)進(jìn)行對比處理的時候，使用map特性的方法時間消耗大概在40毫秒左右，但是其他方法要將近10秒鐘，

所以在對于大量數(shù)據(jù)的處理過程中，還是非常建議使用這種方法的。

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;
}