如何使用流将元素映射到其索引？

Question

我得到了一些自定义对象的流，我想创建一个Map<Integer, MyObject>以每个对象的索引为键的映射。给你一个简单的例子：

Stream<String> myStream = Arrays.asList("one","two","three").stream();
Integer i = 0;
Map<Integer, String> result3 = myStream.collect(Collectors.toMap(x -> i++, x -> x));

显然，这不能编译，因为：

从lambda表达式引用的局部变量必须是final或有效的final

有没有一种简单的方法可以将流的元素映射到它们的索引，以使上述示例的预期输出类似于：

{1=one, 2=two, 3=three}

Answer 1

您可以使用IntStream解决此问题：

List<String> list = Arrays.asList("one","two","three");
Map<Integer, String> map = IntStream.range(0, list.size()).boxed()
        .collect(Collectors.toMap(Function.identity(), list::get));

您创建一个IntStream从0到list.size() - 1（IntStream.range()从流中排除最后一个值）并将每个索引映射到列表中的值。该解决方案的优势在于，它也可以与并行流一起使用，而使用不能实现AtomicInteger。

因此，这种情况下的结果将是：

{0=one, 1=two, 2=three}

要开始第一个索引，1您可以1在收集期间添加：

List<String> list = Arrays.asList("one", "two", "three");
Map<Integer, String> map = IntStream.range(0, list.size()).boxed()
        .collect(Collectors.toMap(i -> i + 1, list::get));

这将导致：

{1=one, 2=two, 3=three}

Answer 2

一个不需要随机访问源数据的干净解决方案是

Map<Integer,String> result = Stream.of("one", "two", "three")
    .collect(HashMap::new, (m,s) -> m.put(m.size() + 1, s),
        (m1,m2) -> {
            int offset = m1.size();
            m2.forEach((i,s) -> m1.put(i + offset, s));
        });

这也适用于并行流。

在不太可能的情况下，这是重复执行的任务，值得将逻辑放入可重用的收集器中，包括一些优化：

public static <T> Collector<T,?,Map<Integer,T>> toIndexMap() {
    return Collector.of(
        HashMap::new,
        (m,s) -> m.put(m.size() + 1, s),
        (m1,m2) -> {
            if(m1.isEmpty()) return m2;
            if(!m2.isEmpty()) {
                int offset = m1.size();
                m2.forEach((i,s) -> m1.put(i + offset, s));
            }
            return m1;
        });
}

然后可以像

Map<Integer,String> result = Stream.of("one", "two", "three")
    .collect(MyCollectors.toIndexMap());

要么

Map<Integer,Integer> result = IntStream.rangeClosed(1, 1000)
    .boxed().parallel()
    .collect(MyCollectors.toIndexMap());

Answer 3

您的i变量实际上不是最终的。

您可以AtomicInteger用作Integer包装器：

Stream<String> myStream = Arrays.asList("one","two","three").stream();
AtomicInteger atomicInteger = new AtomicInteger(0);
Map<Integer, String> result3 = myStream.collect(Collectors.toMap(x -> atomicInteger.getAndIncrement(), Function.identity()));

我认为这有点棘手，因为它只能解决有效的最终变量问题。由于它是特殊的ThreadSafe版本，因此可能会带来一些开销。塞缪尔·菲利普（Samuel Philipp）stream的答案中的纯解决方案可能会更好地满足您的需求。