如何使用Java流API获取所有重复元素的索引

Question

如何使用Java流API获取所有重复元素的索引？

例如，我们有：

List<Integer> input =
        new ArrayList<>(Arrays.asList(312, 312, 195, 195, 195, 851, 335, 851, 99));

我可以将类似的元素分组：

result = input.stream().collect(Collectors.groupingBy(input::indexOf))
//result = {0=[312, 312], 2=[195, 195, 195], 5=[851, 851], 6=[335], 8=[99]}

但是我想得到这样的结果：

{312=[0, 1], 195=[2, 3, 4], 851=[5, 7]}

或者这样：

(0,1,2,3,4,5,7)

我该怎么做？

Answer 1

您可以使用intstream

Map<Integer, List<Integer>> map = IntStream.range(0, input.size())
                .boxed()
                .collect(Collectors.groupingBy(i -> input.get(i), HashMap::new,
                        Collectors.toCollection(ArrayList::new)));

// output : {99=[8], 851=[5, 7], 195=[2, 3, 4], 312=[0, 1], 335=[6]}

拥有所有值及其索引后，您可以简单地滤除具有超过1个值的条目。

完整示例：

Map<Integer, List<Integer>> map = IntStream.range(0, input.size())
                .boxed()
                .collect(Collectors.groupingBy(i -> input.get(i), HashMap::new,
                        Collectors.toCollection(ArrayList::new)))
                .entrySet()
                .stream()
                .filter(e -> e.getValue().size()>1)
                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

Answer 2

使用collections.fquency的另一种方法

int[] dupIndexes = IntStream.range(0, input.size())
                            .filter(i -> Collections.frequency(input, input.get(i)) > 1)
                            .toArray();

Answer 3

看起来仅通过循环就很容易解决。只需将map以输入值作为键，它是索引作为映射的值：

Set<Integer> result = new HashSet<>();
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < input.size(); i++) {
    if (map.containsKey(input.get(i))) {
        result.add(map.get(input.get(i)));
        result.add(i);
    } else {
        map.put(input.get(i), i);
    }
}
System.out.println(result);

输出

[0, 1, 2, 3, 4, 5, 7]

Answer 4

您可以尝试以下代码 -

     List<Integer> input = new ArrayList<>(Arrays.asList(312, 312, 195, 195, 195, 851, 335, 851, 99));
    result = IntStream.range(0, input.size()).boxed()
            .collect(Collectors.groupingBy(input::get, HashMap::new,
                    Collectors.toList()));

Answer 5

    HashMap<Integer, List<Integer>> map = new HashMap<Integer, List<Integer>>();
    List<Integer> ls = null;
    for (int i = 0; i < input.size(); i++) {
        if (map.get(input.get(i)) != null) {
            ls = map.get(input.get(i));
        } else {
            ls = new ArrayList<Integer>();
        }
        ls.add(i);
        map.put(input.get(i), ls);
    }
    System.out.println(map);