使用流API查找列表中项目的所有索引

时间:2014-10-03 12:32:56

标签: java list lambda java-8

我正在尝试使用Java 8流和lambda表达式进行顺序搜索。这是我的代码

List<Integer> list = Arrays.asList(10, 6, 16, 46, 5, 16, 7);
int search = 16;
list.stream().filter(p -> p == search).forEachOrdered(e -> System.out.println(list.indexOf(e)));
Output: 2
        2

我知道list.indexOf(e)始终打印第一次出现的索引。如何打印所有索引?

2 个答案:

答案 0 :(得分:27)

首先,使用Lambdas不是所有问题的解决方案......但是,即便如此,作为for循环,您也可以编写它:

List<Integer> results = new ArrayList<>();
for (int i = 0; i < list.size(); i++) {
    if (search == list.get(i).intValue()) {
        // found value at index i
        results.add(i);
    }
}

现在,没有什么特别的错误,但请注意,这里的关键方面是指数,而不是价值。索引是'loop'的输入和输出。

作为流::

List<Integer> list = Arrays.asList(10, 6, 16, 46, 5, 16, 7);
int search = 16;
int[] indices = IntStream.range(0, list.size())
                .filter(i -> list.get(i) == search)
                .toArray();
System.out.printf("Found %d at indices %s%n", search, Arrays.toString(indices));

产生输出:

Found 16 at indices [2, 5]

答案 1 :(得分:1)

要在 List 中找到 every 值的索引作为 Map,我们可以使用 IntStream 的索引和 Collectors.groupingBy

import java.util.stream.Collectors;
import java.util.stream.IntStream;
//...
final Map<Integer, List<Integer>> indexMap = IntStream.range(0, list.size()).boxed()
        .collect(Collectors.groupingBy(list::get));
//{16=[2, 5], 5=[4], 6=[1], 7=[6], 10=[0], 46=[3]}
//Map of item value to List of indices at which it occurs in the original List

Demo

现在,如果您想获得 search 的索引列表,您可以按如下方式进行:

System.out.println(indexMap.get(search));