好的,阿杰,我用 大白话 + 图解思路帮你一眼看穿 Java/Scala/Stream 中 map 和 flatMap 的区别


map vs flatMap 一眼看懂

1️⃣ 核心概念

函数功能输出结果
map一对一映射每个元素映射成一个新元素
flatMap一对多映射 + 扁平化每个元素映射成一个集合/流,最终扁平成一个流

2️⃣ 通俗理解

  • map:把每个元素“变成别的东西”,结果是原来的大小一样

比喻:每个人穿上不同衣服,人数不变

  • flatMap:把每个元素“变成一个集合”,然后把所有集合合成一个大集合

比喻:每个人可能有多个爱好,把所有人的爱好合成一个列表


3️⃣ Java Stream 示例

示例数据

List<String> names = Arrays.asList("Alice", "Bob");

map 示例

List<Integer> nameLengths = names.stream()
    .map(String::length) // 每个名字映射成长度
    .collect(Collectors.toList());

System.out.println(nameLengths); // [5, 3]

  • 输入:["Alice", "Bob"]
  • 输出:[5, 3]
  • 元素一对一映射

flatMap 示例

假设每个人有多个爱好:

Map<String, List<String>> hobbies = new HashMap<>();
hobbies.put("Alice", Arrays.asList("reading","swimming"));
hobbies.put("Bob", Arrays.asList("running","gaming"));

List<String> allHobbies = hobbies.values().stream()
    .flatMap(List::stream) // 每个 List 扁平化成一个 Stream
    .collect(Collectors.toList());

System.out.println(allHobbies); // [reading, swimming, running, gaming]

  • 输入:[["reading","swimming"], ["running","gaming"]]
  • 输出:["reading","swimming","running","gaming"]
  • 元素一对多映射 + 扁平化

4️⃣ 总结规律

方法输入映射结果最终 Stream
mapTRStream
flatMapTStreamStream(扁平化)
  • 一句话

map 得到的是“流里套流”,flatMap 得到的是“平铺的流”


5️⃣ 小技巧记忆法

  1. map = 一变一:元素个数不变
  2. flatMap = 扁平 + 展开:元素可以变多
  3. 思路:想象每个元素被打开,如果每个元素里还有多个东西,用 flatMap 扁平化

如果你愿意,我可以帮你画一张 map vs flatMap 流程图,直观展示:

[元素1, 元素2] 
   map -> [结果1, 结果2]
   flatMap -> [结果1a, 结果1b, 结果2a, 结果2b]