23

我有以下代码，并想使用 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());
4

5 回答 5

52

这里的基本观察是您的问题涉及非同构转换：单个输入元素可能映射到零、一个或两个输出元素。每当你注意到这一点时，你应该立即开始寻找一个解决方案，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());

（假设import static java.util.stream.Collectors.toList）

于 2015-07-25T18:17:27.837 回答
5

您可以为 lambda 声明一个主体。例如：

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

可能

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

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

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

    if(num == 5) {
        System.out.println("Hey");
    }
};
于 2015-07-25T17:44:20.787 回答
3

当您尝试将flatMap元素添加到管道或一对多映射中时使用。Map 是一对一的映射。

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
于 2015-07-25T18:42:15.920 回答
1

你可以这样做：

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());
于 2015-07-25T18:14:28.600 回答
0

尝试使用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

于 2015-07-25T18:06:51.977 回答