如何获取两个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