21

我希望能够像这样使用 Stream::flatMap

public static List<String> duplicate(String s) {

    List<String> l = new ArrayList<String>();
    l.add(s);
    l.add(s);

    return l;
}


listOfStrings.stream().flatMap(str -> duplicate(str)).collect(Collectors.toList());

但我收到以下编译器错误

Test.java:25：错误：不兼容的类型：无法推断类型变量 R listOfStrings.stream().flatMap(str -> duplicate(str)).collect(Collectors.toList());

（参数不匹配；无法将 lambda 表达式列表中的错误返回类型转换为 Stream）
其中 R，T 是类型变量： R 扩展方法 flatMap(Function>) 中声明的对象 T 扩展接口 Stream 中声明的对象

在scala中，我可以做我认为等效的事情

scala> List(1,2,3).flatMap(duplicate(_))
res0: List[Int] = List(1, 1, 2, 2, 3, 3)

为什么这不是 java 中 flatMap 的有效用法？

4

2 回答 2

28

中的 lambda 表达式flatMap需要返回 a ，从其参数类型Stream可以看出。flatMapFunction<? super T, ? extends Stream<? extends R>>

以下代码将编译并运行良好：

listOfStrings.stream()
             .flatMap(str -> duplicate(str).stream()) // note the .stream() here
             .collect(Collectors.toList());

因为 lambda 表达式str -> duplicate(str).stream()的类型是Function<String, Stream<String>>.

于 2015-09-11T08:13:54.917 回答
8

如果你想多次复制流中的每个对象，你不需要在这上面加上额外的ArrayList. 有几种更短、更快的替代方案。

  • 使用 生成新流Stream.generate，然后对其进行限制：

    listOfStrings.stream()
             .flatMap(str -> Stream.generate(() -> str).limit(2))
             .collect(Collectors.toList());

  • 通过生成数字序列IntStream.range并将它们映射到相同的字符串：

    listOfStrings.stream()
             .flatMap(str -> IntStream.range(0, 2).mapToObj(i -> str))
             .collect(Collectors.toList());

  • 使用好旧的Collections.nCopies：

    listOfStrings.stream()
             .flatMap(str -> Collections.nCopies(2, str).stream())
             .collect(Collectors.toList());

如果您确定您将始终准确复制两次，则有最短的选择：

listOfStrings.stream()
             .flatMap(str -> Stream.of(str, str))
             .collect(Collectors.toList());
于 2015-09-11T09:46:27.890 回答