Java8：当 forEach 须要索引

在 上一篇文章 中，咱们讨论了如何使用 Java8 中 Map 添加的新方法 computeIfAbsent 来统计集合中每一个元素出现的全部位置，代码以下：

public static Map<String, List<Integer>> getElementPositions(List<String> list) {
    Map<String, List<Integer>> positionsMap = new HashMap<>();

    for (int i = 0; i < list.size(); i++) {
        positionsMap.computeIfAbsent(list.get(i), k -> new ArrayList<>(1)).add(i);
    }

    return positionsMap;
}
复制代码

至少有两点须要探讨： 一、若是 list 不是基于数组的（即不是 RandomAccess 的），而是基于链表的，那么 list.get(int index) 方法的效率就值得思考了； 二、既然都有了 Lambda（即当前平台为 Java8），咱们为何还要一次次去写传统的 for 循环呢？

在 Java8 中，为 Iterable 接口添加了默认的 forEach 方法：

很好理解，遍历当前 Iterable 中全部的元素，使用每一个元素做为参数调用一次 action。而 Collection 接口继承了 Iterable 接口，因此全部的继承自 Collection 的集合类均可以直接调用 forEach 方法。好比：

public static void main(String[] args) throws Exception {
    List<String> list = Arrays.asList("a", "b", "b", "c", "c", "c", "d", "d", "d", "f", "f", "g");
    
    list.forEach(str -> System.out.print(str + " "));
    
    System.out.println();
}

复制代码

运行结果：

那若是咱们在遍历的时候须要使用到元素的索引呢（相似 getElementPositions 方法那样）？ 很惋惜，Java8的 Iterable 并无提供一个带索引的 forEach 方法。不过本身动手，丰衣足食——让咱们本身写一个带索引的 forEach 方法：

import java.util.Objects;
import java.util.function.BiConsumer;

/**
 * Iterable 的工具类
 */
public class Iterables {

    public static <E> void forEach(
            Iterable<? extends E> elements, BiConsumer<Integer, ? super E> action) {
        Objects.requireNonNull(elements);
        Objects.requireNonNull(action);

        int index = 0;
        for (E element : elements) {
            action.accept(index++, element);
        }
    }
}
复制代码

该 forEach 方法第一个参数为要遍历的 Iterable，第二个参数为 BiConsumer。BiConsumer 的输入参数第一个即索引，第二个为元素。

咱们测试下这个 forEach 方法：

public static void main(String[] args) throws Exception {

    List<String> list = Arrays.asList("a", "b", "b", "c", "c", "c", "d", "d", "d", "f", "f", "g");

    Iterables.forEach(list, (index, str) -> System.out.println(index + " -> " + str));
}

复制代码

运行结果：

结果和预期的一致。

如今咱们使用 Iterables.forEach 改写 getElementPositions方法：

public static Map<String, List<Integer>> getElementPositions(List<String> list) {
    Map<String, List<Integer>> positionsMap = new HashMap<>();

    Iterables.forEach(list, (index, str) -> {
        positionsMap.computeIfAbsent(str, k -> new ArrayList<>(1)).add(index);
    });

    return positionsMap;
}

public static void main(String[] args) throws Exception {
    List<String> list = Arrays.asList("a", "b", "b", "c", "c", "c", "d", "d", "d", "f", "f", "g");

    System.out.println("使用 computeIfAbsent 和 Iterable.forEach：");
    Map<String, List<Integer>> elementPositions = getElementPositions(list);
    System.out.println(elementPositions);
}
复制代码

运行结果和原来一致：

真的不明白这么简单且实用的方法，Java8 为何不在 Iterable 中提供一个默认实现（此处应有黑人问号）。

原文做者：mizhoux 原文地址：segmentfault.com/a/119000000…

推荐

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

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

文末

欢迎关注我的微信公众号：Coder编程 欢迎关注Coder编程公众号，主要分享数据结构与算法、Java相关知识体系、框架知识及原理、Spring全家桶、微服务项目实战、DevOps实践之路、每日一篇互联网大厂面试或笔试题以及PMP项目管理知识等。更多精彩内容正在路上~ 新建了一个qq群：315211365，欢迎你们进群交流一块儿学习。谢谢了！也能够介绍给身边有须要的朋友。

文章收录至 Github: github.com/CoderMerlin… Gitee: gitee.com/573059382/c… 欢迎关注并star~