flatMap 三重嵌套流

Question

我正在尝试 flatMap 并从三个列表中获取字符串列表的结果。我以某种方式能够通过以下代码做到这一点。代码正在运行，但不知何故我觉得我把它弄得太复杂了。有人可以给我一些建议，可以更好地改进它

countries.stream()
    .map(country -> titles.stream()
        .map(title -> games.stream()
            .map(game -> Arrays.asList(game, country + "_" + title + "_" + game))
            .collect(toList()))
        .collect(toList()))
    .collect(toList())
    .stream()
    .flatMap(Collection::stream)
    .flatMap(Collection::stream)
    .flatMap(Collection::stream)
    .collect(Collectors.toSet());

为了澄清逻辑，传统方法如下所示：

Set<List<String>> results = new HashSet<>();
for (String country : countries) {
    for (String title : titles) {
        for (String game : games) {
            results.add(Arrays.asList(game, country + "_" + title + "_" + game));
        }
    }
}

Answer 1

您可以分两步完成：

首先创建国家和标题列表的串联：

List<String> countriesTitle = countries.stream()
            .flatMap(country -> titles.stream()
                    .map(title -> country + "_" + title))
            .collect(Collectors.toList());

然后根据之前的结果创建连接列表country+"_"+title+"_"+game字符串：

Stream.concat(games.stream(),
                    games.stream()
                          .flatMap(game -> countriesTitle.stream()
                                .map(countryTitle -> countryTitle + "_" + game)))
         .collect(Collectors.toList());

---

更新答案：

games.stream()
      .flatMap(game -> countries.stream()
               .flatMap(country -> titles.stream()
                   .flatMap(title -> Stream.of(game, country + "_" + title + "_" + game))))
       .collect(Collectors.toSet());