Skip to main content
 首页 » 编程设计

java实现获取两个list交集

2022年07月19日133leader

如何获取两个list的交集,使用java8 的stream实现非常容易。

java流实现

创建两个字符串list,两者都有重复的元素:

List<String> list = Arrays.asList("red", "blue", "blue", "green", "red"); 
List<String> otherList = Arrays.asList("red", "green", "green", "yellow"); 

下面使用流方式获取交集:

Set<String> result = list.stream() 
  .distinct() 
  .filter(otherList::contains) 
  .collect(Collectors.toSet()); 
 
Set<String> commonElements = new HashSet(Arrays.asList("red", "green")); 
 
Assert.assertEquals(commonElements, result); 

首先,删除重复元素,然后使用filter选择另一个集合也包含的元素。最后使用Collector转换输出,交集应该包括的公共元素仅以此。如果顺序不重要,可以直接使用toSet,但也可以toList或其他collector方法。

自定义类集合求交集

如果集合中存放的不是字符串,而是用户自定义类呢?只要我们遵从java规范,基于流的解决方案同样适用于自定义类。
判断集合是否包含特定对象,有equals方法决定。因此必须重写equals方法,确保比较两个对象是基于它们的一些属性值。

举例,基于宽和高比较两个矩阵。如果没有重写equals方法,类会适用父类equals的实现。一致追溯继承链至Object类的equals方法,如果它们引用堆中相同的对象则两个实例相等。

总结

本文介绍了如何计算两个集合的交集。其他方法实现会比较繁琐,利用流API方式实现非常简单直接。


本文参考链接:https://blog.csdn.net/neweastsun/article/details/115359819
阅读延展