Java Intstream 获取具有空值的多维数组的索引

Question

Java Instream 我需要具有空值的二维数组中的元素索引。我可以使用 对于我在下面做的一维数组，它起作用了：

  IntStream.range(0, myArr.length)
                .filter(x -> myArr[x] == null)
                .findFirst()
                .orElse(-1);

我可以使用 Streams over Arrays 和 flatmap 获取对象，但我的目的是获取 i 和 j 的值，其中 myArr[i][j] 为空。

Answer 1

您可以使用一个long 值来携带两个int 值：

long both = LongStream.range(0, array.length)
    .flatMap(i -> LongStream.range(0, array[(int)i].length).map(j -> j << 32 | i))
    .filter(l -> array[(int)l][(int)(l>>32)] == null)
    .findFirst().orElse(-1);

int j = (int) (both >> 32), i = (int)both;

if(j >= 0) {
    System.out.println("found ("+i+", "+j+")  "+array[i][j]);
}

但这并不比例如简单

int i, j;
search: {
    for(i = 0; i < array.length; i++) {
        var sub = array[i];
        for(j = 0; j < sub.length; j++)
            if(sub[j] == null) break search;
    }
    j = -1;
}

if(j >= 0) {
    System.out.println("found ("+i+", "+j+")  "+array[i][j]);
}

Answer 2

您可以使用Stream.iterate 和flatMap 将值为null 的所有索引放在SimpleEntry 中，其中键是第一个索引，值是第二个索引：

 Integer[][] integers = {{1, 2, 3, 6}, {4, 5, 6}, {7, 8, null, 11}};
 List<SimpleEntry<Integer, Integer>> collect = 
            Stream.iterate(0, (i) -> i < integers.length, (i) -> ++i)
                .flatMap(i -> IntStream.range(0, integers[i].length)
                      .filter(j -> integers[i][j] == null)
                      .mapToObj(j -> new SimpleEntry<Integer, Integer>(i, j))) 
               .collect(Collectors.toList());

 System.out.println(collect);

那么对于collect 列表中的所有条目integers[entry.getKey()][entry.getValue()] == null 是true

编辑： 问题本身是微不足道的，唯一要做的决定是应该使用什么容器来存储索引。这是Integerarrays 的“自然”解决方案：

    List<Integer[]> collect1 = new ArrayList<>();
    for (int i = 0; i < integers.length; i++){
        for (int j = 0; j < integers[i].length; j++){
            if (integers[i][j] == null){
                collect1.add(new Integer[]{i, j});
            }
        }
    }

    collect1.forEach(arr -> System.out.println("null at " +arr[0] + "," + arr[1]));