java8中的map与flatmap区别

偏执的太偏执、 2023-10-01 08:51 14阅读 0赞

map:只能返回一个值

flatmap:返回多个值

new ArrayList().stream().map(x -> x);//返回一个  
new ArrayList().stream().flatMap(x -> Arrays.asList(x.split(" ")).stream());//返回一个流,也就是多个值

看API声明可以发现，flatmap接受的参数是流类型，map接受的非流类型，flatmap会把结果中的所有流的所有元素合并起来。

## 问题 ##

假如我们有这样一个需求给定单词列表\[“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方法示意图：

![img][]

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

flatMap方法示意图：

![img][img 1]

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

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

![img][img 2]

## 解决问题： ##

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

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的区别了

[img]: https://img-blog.csdnimg.cn/20201206204241203.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMwNDU0Mzc=,size_16,color_FFFFFF,t_70
[img 1]: https://img-blog.csdnimg.cn/20201206204425675.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMwNDU0Mzc=,size_16,color_FFFFFF,t_70
[img 2]: https://img-blog.csdnimg.cn/img_convert/3f74ff613ab866a99c51f9dce4da366d.png