Question

我有以下代码，并希望使用lambda函数实现它只是为了好玩。可以使用基本的聚合操作来完成吗？

List<Integer> result = new ArrayList<>();

for (int i = 1; i <= 10; i++) {
    if (10 % i == 0) {
        result.add(i);
        if (i != 5) {
            result.add(10 / i);
        }
    }
}

使用lambda：

List<Integer> result = IntStream.rangeClosed(1, 10)
                                .boxed()
                                .filter(i -> 10 % i == 0)
                                // a map or forEach function here?
                                // .map(return 10 / i -> if i != 5)
                                .collect(Collectors.toList());

Answer 1

这里的基本观察是你的问题涉及非同构变换：单个输入元素可以映射到零个，一个或两个输出元素。每当您注意到这一点时，您应该立即开始寻找涉及flatMap而不是map的解决方案，因为这是实现这种一般转换的唯一方法。在您的特定情况下，您可以先将filter应用于一对一元素映射，然后flatMap进行一对二映射：

List<Integer> result =
    IntStream.rangeClosed(1, 10)
             .filter(i -> 10 % i == 0)
             .flatMap(i -> i == 5 ? IntStream.of(i) : IntStream.of(i, 10 / i))
             .boxed()
             .collect(toList());

Answer 2

你可以为lambda声明一个body。例如：

Runnable run = () -> System.out.println("Hey");

可能是

Runnable run = () -> {
    System.out.println("Hey");
};

在该主体中，您可以创建嵌套语句：

Runnable run = () -> {
    int num = 5;

    if(num == 5) {
        System.out.println("Hey");
    }
};

Answer 3

在尝试将元素添加到管道或1对多映射时使用flatMap。地图是一对一的映射。

ArrayList<Integer> result = (ArrayList<Integer>) IntStream.rangeClosed(1, 10)
                .boxed()
                .filter(i -> 10 % i == 0)
                .flatMap((Integer i) -> {return i!=5 ? Stream.of(i, (10/i)):Stream.of(i);})
                .collect(Collectors.toList());

这导致与

相同的列表

ArrayList<Integer> result2 = new ArrayList<Integer>();

        for (int i = 1; i <= 10; i++) {
            if (10 % i == 0) {
                result2.add(i);
                if (i != 5) {
                    result2.add(10 / i);
                }
            }
        }

如果您想知道哪种方式更快，循环方法比使用流快〜3倍。

Benchmark                     Mode  Cnt      Score     Error  Units
testStreams.Bench.loops       avgt    5     75.221 ±   0.576  ns/op
testStreams.Bench.streams     avgt    5    257.713 ±  13.125  ns/op

Answer 4

你可以这样做：

List<Integer> result1 = IntStream
    .rangeClosed(1, 10)
    .boxed()
    .filter(i -> 10 % i == 0)
    .map(i -> (i != 5 ? Stream.of(i, 10 / i) : Stream.of(i)))
    .flatMap(Function.identity())
    .collect(Collectors.toList());

Answer 5

尝试使用flatMap：

List<Integer> result = IntStream.rangeClosed(1, 10)
        .boxed()
        .flatMap((i) -> {
            List<Integer> results = new ArrayList<>();
            if (10 % i == 0) {
                results.add(i);
                if (i != 5) {
                    results.add(10 / i);
                }
            }
            return results.stream();
        })
        .collect(Collectors.toList());

请参阅http://ideone.com/EOBiEP

如何使用Java 8 / lambda执行嵌套的'if'语句？

5 个答案: