multidimensional-array - java8数组流到二维数组

Question

我是 Java8 的新手，无法使用流将一个数组映射到另一个二维数组。

我有一个二维数组，它是一种模式:

boolean[][] pattern = {
            {true, true, false},
            {true, false, true},
            {false, true, true}
    };

和包含键的第二个数组。

0 表示:从模式中取 0 个元素

1 表示:从模式中取 1 个元素，依此类推

int[] keys = {2, 1, 0};

从这两个数组中，我想生成另一个二维数组。在这种情况下，结果将如下所示:

boolean[][] result = {
                {false, true, true},
                {true, false, true},
                {true, true, false}
        };

这是Java7中的代码:

public boolean[][] producePlan(int[] keys, boolean[][] pattern) {
        boolean[][] result = new boolean[keys.length][];
        for (int i = 0; i < keys.length; i++) {
            result[i] = pattern[keys[i]];
        }
        return result;
    }

在 Java8 中，我只能打印每一行

Arrays.stream(keys).mapToObj(x -> pattern[x]).forEach(x -> System.out.println(Arrays.toString(x)));

但不能将其转换为二维数组。

请帮忙

Answer 1

你可以这样做，

boolean[][] result = Arrays.stream(keys).mapToObj(i -> pattern[i]).toArray(boolean[][]::new);

既然你有 Stream<boolean[]>后 map stage，你只需要提供一个数组生成器函数。