java - 有没有办法在 Java 8 中执行 Ruby every_slice ？

Question

如果我有一个数组

[1, 2, 3, 4, 5, 6, 7, 8, 9]

并且想要将其分成 3 组，我可以使用 .each_slice(3) 如下

[1, 2, 3, 4, 5, 6, 7, 8, 9].each_slice(3).to_a => [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

有没有办法可以在 Java 8 中做类似的事情？

Ruby enumerable each_slice

Answer 1

我想出了这个解决方案:

public static int[][] slice(int[] arr, int n) {
    return IntStream.range(0, (int) Math.ceil((double) arr.length / n))
            .mapToObj(i -> Arrays.copyOfRange(arr, i * n, Math.min((i + 1) * n, arr.length)))
            .toArray(int[][]::new);
}