java实现获取两个list交集

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

java流实现

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

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

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

Set result = list.stream()
  .distinct()
  .filter(otherList::contains)
  .collect(Collectors.toSet());

Set commonElements = new HashSet(Arrays.asList("red", "green"));

Assert.assertEquals(commonElements, result);

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

自定义类集合求交集

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

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

总结

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

你可能感兴趣的:(java8~9核心功能)