使用Java 8 Stream在2个列表中查找匹配的元素


问题内容

我的情况是:

class Person {
    String id ;
    String name;
    String age;
}
List<Person> list1 = {p1,p2, p3};
List<Person> list2 = {p4,p5, p6};

我想知道是否有list1一个名字和年龄相同list2但不在乎的人id

什么是最快的方法?


问题答案:

为自己定义一个关键对象,该对象可以保存并比较所需的属性。在这种简单情况下,您可以使用一个小的列表,而每个索引对应一个属性。对于更复杂的情况,可以使用Map(使用属性名称作为键)或专用类:

Function<Person,List<Object>> toKey=p -> Arrays.asList(p.getName(), p.getAge());

具有这种映射功能。您可以使用简单的解决方案:

list1.stream().map(toKey)
     .flatMap(key -> list2.stream().map(toKey).filter(key::equals))
     .forEach(key -> System.out.println("{name="+key.get(0)+", age="+key.get(1)+"}"));

当您的列表很大时,这可能会导致性能不佳。如果列表很大(或者无法预测它们的大小),则应该使用中间变量Set来加速查找(将任务的时间复杂度从更改O(n²)O(n)):

list2.stream().map(toKey)
     .filter(list1.stream().map(toKey).collect(Collectors.toSet())::contains)
     .forEach(key -> System.out.println("{name="+key.get(0)+", age="+key.get(1)+"}"));

在上面的示例中,每个匹配项都被打印出来。如果您仅对是否存在这样的匹配感兴趣,则可以使用以下任一方法:

boolean exists=list1.stream().map(toKey)
     .anyMatch(key -> list2.stream().map(toKey).anyMatch(key::equals));

要么

boolean exists=list2.stream().map(toKey)
     .anyMatch(list1.stream().map(toKey).collect(Collectors.toSet())::contains);