Java将二维数组转换为一维数组

Question

我正在尝试将二维数组转换为 Java 中的一维数组。 我不能使用任何额外的数据结构、ArrayLists、集合等作为临时存储。 只能分配一个数组 - 将从该方法返回的数组。

共有三个测试，其中将数组输入到方法中以验证程序。 测试一有一个 3 行和列的二维数组，测试二有一个 3 行 1 列的二维数组，测试三为空。

我目前的代码是：

public static int[] twoDConvert(int[][] nums) {
    int index = 0;
    int[] combined = new int[nums.length*3];
    if (nums.length == 0) return combined;  
        for (int row = 0; row < nums.length; row++) {
            for (int col = 0; col < nums.length; col++) {
                combined[index] += nums[row][col];
                index++;
        }
    }
    return combined;
}

第一个测试工作正常，但由于某种原因，第二个测试抛出 ArrayIndexOutOfBoundsException（长度为 1 的索引 1 越界）。 无论组合数组长度有多大或多小，都会发生这种情况。 我怎样才能解决这个问题？

Answer 1

我们创建一个长度由 size 方法确定的 Integer 数组。 Size 获取二维数组并找到每行中所有列的总和并返回该总和。 我们检查长度是否为 0，如果是，则返回。 然后我们循环遍历所有行，然后循环遍历该行中的所有列。 然后我们将该行和列的值分配给数组中的下一个索引。 不幸的是，我们无法做任何数学运算（据我所知）来确定基于行/列的索引，因为它们可能是可变大小。

    public static int[] twoDConvert(int[][] nums) {
        int[] combined = new int[size(nums)];

        if (combined.length <= 0) {
            return combined;
        }
        int index = 0;

        for (int row = 0; row < nums.length; row++) {
            for (int column = 0; column < nums[row].length; column++) {
                combined[index++] = nums[row][column];
            }
        }
        return combined;
    }

    private static int size(int[][] values) {
        int size = 0;

        for (int index = 0; index < values.length; index++) {
            size += values[index].length;
        }
        return size;
    }