Java8 中 Map 接口的新方法 js中map函数的用法

我们提一个需求：给定一个 List<String>，统计每个元素出现的所有位置。

比如，给定 list：["a", "b", "b", "c", "c", "c", "d", "d", "d", "f", "f", "g"]，那么应该返回：

a : [0]
b : [1, 2]
c : [3, 4, 5]
d : [6, 7, 8]
f : [9, 10]
g : [11]

很明显，我们很适合使用Map来完成这件事情：

运行结果：

Java8 时，Map<K, V> 接口添加了一个新的方法，putIfAbsent(K key, V value)，功能是：
如果当前 Map不存在键key 或者该 key 关联的值为 null，那么就执行 put(key, value)；否则，便不执行 put 操作。该方法等价于如下代码：

(putIfAbsent方法与put方法一样，返回的是方法调用之前与参数 key 相关联的value）

使用 putIfAbsent 修改 getElementPositions 方法：

运行结果：

可以看到使用 putIfAbsent 之后的 getElementPositions 简洁了一点，那还能更简洁吗？

查看 Map接口的方法，可以发现在 JDK1.8 时，还添加了如下两个方法：

查看compute 方法的API 文档，可以发现compute方法与如下代码等价

compute方法和原来put方法的区别在于：

put(K key, V value) 方法，如果 key 在 Map 中不存在，那么直接加入；如果已经存在，那么使用新的 value 替换旧的value；

而compute(K key, BiFunction remappingFunction)方法可以通过一个 BiFunction 来计算出新的value，BiFunction的参数为旧的 key 和 value，返回计算出新的value —— 与 put方法不同，compute方法返回的会是最新的与key 相关联的 value，而不是旧的 value。
所以可以使用compute 方法改写 getElementPositions 如下：

(k, v) -> v == null ? new ArrayList<>(1) : v即 如果当前的 value 为 null，那么 该 BiFunction的返回值为 new ArrayList<>(1)；如果不为 null，那么返回值便是本身。而且因为compute 方法会返回新的value —— 此时便是与 list.get(i)（key） 相关联的ArrayList —— 所以我们可以直接调用其add 方法。

运行结果：

使用 compute 的运行结果

很棒~ 还能更简洁吗？ 我们再看看 computeIfAbsent 方法：
computeIfAbsent 和 compute 的关系，就类似于 putIfAbsent 和put的关系：
computeIfAbsent在 key不在Map 中或者与key 相关联的value为null时，才执行通过函数计算新 value 的操作，否则不执行；computeIfAbsent的返回值也是与 key 相关联的最新的 value。其默认实现如下：

computeIfAbsent 的默认实现

与 compute 不同，computeIfAbsent接受的函数操作是 Function 而不是 BiFunction—— 这很好理解，computeIfAbsent 只在key 不在 Map 中或者与 key相关联的 value 为null 时才执行函数操作，那么显然此时与 key相关的 value为 null，所以computeIfAbsent只接受 Function作为参数即可 —— 该 Function可以使用key作为参数计算出新的value。使用 computeIfAbsent 改写 getElementPositions：

运行结果：

事实上，本文使用 putIfAbsent 时是存在问题的，positionsMap.putIfAbsent(str, new ArrayList<>(1)); 这句代码每次调用时都会产生一个临时的 ArrayList—— 当遍历的List<String>较大时，这可能会带来一定的负面影响；相比之下 compute和 computeIfAbsent 的好处在于，它们接受的参数为函数，只会在必要时才使用函数进行计算得出新value。在本文类似需求的情况下，就适用性和简洁性而言，computeIfAbsent要优于compute。在 JDK1.8 的 API 文档中，也说到在需要生成一个类似于 Map<K, Collection<V>> 的结构时，computeIfAbsent 很适合这种情况：

那 compute方法适用于什么情况呢？从前面的介绍可知，compute 方法更适用于更新 key关联的 value时，新值依赖于旧值的情况 —— 比如统计一个 List<String> 中每个元素出现的次数：

运行结果：

Java8中还为Map添加了一些其他方便于编码的新方法，请有兴趣的读者继续发掘。

原文地址：https://segmentfault.com/a/1190000007838166

推荐

大厂笔试内容集合（内有详细解析） 持续更新中….

ProcessOn是一个在线作图工具的聚合平台~

文末

欢迎关注个人微信公众号：Coder编程
欢迎关注Coder编程公众号，主要分享数据结构与算法、Java相关知识体系、框架知识及原理、Spring全家桶、微服务项目实战、DevOps实践之路、每日一篇互联网大厂面试或笔试题以及PMP项目管理知识等。更多精彩内容正在路上~