java8中的map与flatmap

朱雀 2022-12-27 01:20 150阅读 0赞

# 问题 #

假如我们有这样一个需求给定单词列表\["Hello","World"\]，你想要返回列表\["H","e","l", "o","W","r","d"\],如果我们使用java实现这个需求，你会怎么实现呢？可能我们第一个想法是下面这种写法：

words.stream()
    .map(word -> word.split(""))
    .distinct()
    .collect(toList());

但是如果我们运行会发现这样的写法是不正确的，这个方法的问题在于，传递给map方法的Lambda为每个单词返回了一个String\[\]（String列表）。因此， map 返回的流实际上是Stream 类型的。你真正想要的是用  
Stream来表示一个字符流。因此，这是行不通的。

我们带着上面的问题进行下面的分析：

# map和flatmap的区别 #

map:对于stream中包含的元素使用 特定的转换函数进行转换操作，新生成的Stream只包含转换生成的元素。这个方法有三个对于原始类型的变种方法，分别是mapToInt,mapToLong和mapToDouble.这三个方法比较好理解，比如mapToInt就是把原始的Stream转换成一个新的Stream，这个新生成的Stream中的元素map方法示意图：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMwNDU0Mzc_size_16_color_FFFFFF_t_70][]

flatMap：和map类似，不同的是每个元素 转换得到是Stream对象，会把子Stream中的元素压缩到父集合中:

flatMap方法示意图：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMwNDU0Mzc_size_16_color_FFFFFF_t_70 1][]

如果我们用一句话来总结flatmap：

把几个小的list转换成一个大的list

![3f74ff613ab866a99c51f9dce4da366d.png][]

# 解决问题： #

对于我们开始提出的问题，我们分为两步来完成：

1、分隔字符串，将数据中的字母都分隔开，形成多个小list新的stream

2、组装成新的list

String[] strings = {"Hello", "World"};
            List<String> collect = Stream.of(strings)
                    .map(s -> s.split(""))
                    .flatMap(s -> Stream.of(s))
                    .distinct()
                    .collect(Collectors.toList());
    
            collect.forEach(c->{
                System.out.println(c);
            });

这样我们就可以很好地理解map和flatmap的区别了

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMwNDU0Mzc_size_16_color_FFFFFF_t_70]: /images/20221120/2b365e8414b54007b9cfe2697e503c78.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMwNDU0Mzc_size_16_color_FFFFFF_t_70 1]: /images/20221120/a2e6e82355ff47f784f86e330efd8e07.png
[3f74ff613ab866a99c51f9dce4da366d.png]: https://img-blog.csdnimg.cn/img_convert/3f74ff613ab866a99c51f9dce4da366d.png