Java中构建Set元素索引数组及Lambda表达式报错解决

你的需求是：给定一个Integer类型的Set，生成一个数组，数组中对应元素值的位置存储该元素在Set中的索引，比如Set {2,4,5}对应的数组是[null, null, 0, null, 1, 2]。

你当前代码报错的原因是：Lambda表达式中引用的外部局部变量必须是final或有效final（即变量赋值后不再被修改）。而你代码里的index在Lambda中执行了index++的修改操作，不符合这个规则，所以抛出了"Variable used in lambda expression should be final or effectively final"错误。

下面是两种可行的解决方案：

方案一：使用AtomicInteger包装索引 #

利用原子类的引用不可变、内部值可修改的特性，绕过Lambda的变量限制：

import java.util.concurrent.atomic.AtomicInteger;
import java.util.Set;

private Integer[] getIndexArray(Set<Integer> sourceSet) {
    Integer[] resultArray = new Integer[Constants.MAX_IDS];
    AtomicInteger index = new AtomicInteger(0);
    sourceSet.forEach(id -> resultArray[id] = index.getAndIncrement());
    return resultArray;
}

方案二：改用普通for循环 #

直接放弃Lambda，用传统的增强for循环遍历Set，避免Lambda的变量约束：

import java.util.Set;

private Integer[] getIndexArray(Set<Integer> sourceSet) {
    Integer[] resultArray = new Integer[Constants.MAX_IDS];
    int index = 0;
    for (Integer id : sourceSet) {
        resultArray[id] = index++;
    }
    return resultArray;
}

注意事项 #

普通HashSet是无序的，遍历顺序不固定，如果需要索引和元素的顺序严格匹配你的预期，要确保sourceSet是有序集合（比如LinkedHashSet或TreeSet）。

内容的提问来源于stack exchange，提问作者Leah