通过谓词找到第一个元素

我刚刚开始使用Java 8 lambdas，我正在尝试实现一些我在函数式语言中习惯的东西。

例如，大多数函数式语言都有某种find函数，它对序列或列表进行操作，返回第一个元素，其谓词为true。我认为在Java 8中实现这一目标的唯一方法是:

lst.stream()
.filter(x -> x > 5)
.findFirst()

然而，这似乎对我来说效率很低，因为过滤器会扫描整个列表，至少在我的理解(这可能是错误的)。有没有更好的办法?

547888 次浏览
小开
最佳答案

不，过滤器不扫描整个流。它是一个中间操作，返回一个惰性流(实际上所有中间操作都返回一个惰性流)。为了说服你，你可以简单地做下面的测试:

List<Integer> list = Arrays.asList(1, 10, 3, 7, 5);
int a = list.stream()
.peek(num -> System.out.println("will filter " + num))
.filter(x -> x > 5)
.findFirst()
.get();
System.out.println(a);

输出:

will filter 1
will filter 10
10

您可以看到，实际上只处理了流的前两个元素。

所以你可以用你的方法，这是完全没问题的。

小开

然而，这对我来说似乎效率很低，因为过滤器会扫描整个列表

不，它不会——一旦找到满足谓词的第一个元素，它就会“中断”。你可以在流包javadoc中读到更多关于懒惰的内容，特别是(强调我的):

许多流操作(如过滤、映射或重复删除)可以延迟实现，从而为优化提供了机会。例如，“找到带有三个连续元音的第一个字符串”不需要检查所有输入字符串。流操作分为中间(产生流)操作和终端(产生价值或副作用)操作。# EYZ0

小开 
return dataSource.getParkingLots()
.stream()
.filter(parkingLot -> Objects.equals(parkingLot.getId(), id))
.findFirst()
.orElse(null);

我必须从对象列表中只过滤出一个对象。所以我用了这个，希望能有所帮助。

小开

除了亚历克西斯C的答案，如果您正在使用一个数组列表，其中您不确定您正在搜索的元素是否存在，请使用这个。

Integer a = list.stream()
.peek(num -> System.out.println("will filter " + num))
.filter(x -> x > 5)
.findFirst()
.orElse(null);

然后可以简单地检查一个是否是null。

小开

如果你正在寻找一个布尔返回值，我们可以通过添加isPresent来做得更好:

return dataSource.getParkingLots().stream().filter(parkingLot -> Objects.equals(parkingLot.getId(), id)).findFirst().isPresent();
小开 

import org.junit.Test;


import java.util.Arrays;
import java.util.List;
import java.util.Optional;


// Stream is ~30 times slower for same operation...
public class StreamPerfTest {


int iterations = 100;
List<Integer> list = Arrays.asList(1, 10, 3, 7, 5);




// 55 ms
@Test
public void stream() {


for (int i = 0; i < iterations; i++) {
Optional<Integer> result = list.stream()
.filter(x -> x > 5)
.findFirst();


System.out.println(result.orElse(null));
}
}


// 2 ms
@Test
public void loop() {


for (int i = 0; i < iterations; i++) {
Integer result = null;
for (Integer walk : list) {
if (walk > 5) {
result = walk;
break;
}
}
System.out.println(result);
}
}
}
小开
@AjaxLeung已经回答了，但在评论中很难找到。
只检查

lst.stream()
.filter(x -> x > 5)
.findFirst()
.isPresent()

化简为

lst.stream()
.anyMatch(x -> x > 5)
小开

对于我来说，一个带有循环的通用实用函数似乎要干净得多:

static public <T> T find(List<T> elements, Predicate<T> p) {
for (T item : elements) if (p.test(item)) return item;
return null;
}


static public <T> T find(T[] elements, Predicate<T> p) {
for (T item : elements) if (p.test(item)) return item;
return null;
}

在使用:

List<Integer> intList = Arrays.asList(1, 2, 3, 4, 5);
Integer[] intArr = new Integer[]{1, 2, 3, 4, 5};


System.out.println(find(intList, i -> i % 2 == 0)); // 2
System.out.println(find(intArr, i -> i % 2 != 0)); // 1
System.out.println(find(intList, i -> i > 5)); // null

提交答案