Question

我正在尝试用Java实现Strassen算法，我正处于需要将输出组合成单个矩阵/ 2D数组的步骤。我使用System.arraycopy来复制数组，这对于以自上而下的方式连接两个数组很有效，但是，我还需要将它们并排连接，我遇到了麻烦。我遇到了ArrayOutOfBoundsException。这是我的代码

static int[][] Consolidate(int[][] c11, int[][] c12, int[][] c21, int[][] c22) {
    /* check size compatibility */
    if(c11[0].length == c21[0].length && 
            c11.length == c12.length &&
            c21.length == c22.length &&
            c22[0].length == c12[0].length) {
        int _rowSize = c11.length + c21.length;
        int _colSize = c11[0].length + c12[0].length;
        int[][] retArray = new int[_rowSize][_colSize];

        int[][] ltArray = new int[_rowSize][c11[0].length];
        int[][] rtArray = new int[_rowSize][c12[0].length];

        System.arraycopy(c11, 0, ltArray, 0, c11.length);
        System.arraycopy(c21, 0, ltArray, c11.length, c21.length);
        System.arraycopy(c12, 0, rtArray, 0, c12.length);
        System.arraycopy(c22, 0, rtArray, c12.length, c22.length);

        System.arraycopy(ltArray, 0, retArray, 0, ltArray.length);
        System.arraycopy(rtArray, 0, retArray, ltArray.length, rtArray.length);
        return retArray;
    }
    return null;
}

最后一行

System.arraycopy(rtArray, 0, retArray, ltArray.length, rtArray.length);

抛出异常。有没有办法并排连接数组（以列方式）？

Answer 1

编辑以下是经过验证的答案

您可以手动复制最后一部分。

替换

System.arraycopy(ltArray, 0, retArray, 0, ltArray.length);
System.arraycopy(rtArray, 0, retArray, ltArray.length, rtArray.length);

使用

//Commented both calls
//System.arraycopy(ltArray, 0, retArray, 0, ltArray.length);
//System.arraycopy(rtArray, 0, retArray, ltArray.length, rtArray.length);
for (int row = 0; row < ltArray.length; row++) {
    int colInTarget = 0;
    for (int col = 0; col < ltArray[row].length; col++,colInTarget++) {
        retArray[row][colInTarget] = ltArray[row][col];
    }
    for (int col = 0; col < rtArray[row].length; col++,colInTarget++) {
        retArray[row][colInTarget] = rtArray[row][col];
    }
}

Answer 2

我认为没有办法逐列连接，因为你可以用octave或matlab来实现。

这背后的原因是java支持一维数组。矩阵是一维数组，每个元素也是一个数组（不一定是相同的大小！）。

所以，这是在循环中逐个元素复制的最直接的解决方案。

private static int[][] consolidate(int[][] a, int[][] b, int[][] c, int[][] d) {
        int n = a.length * 2;
        int[][] result = new int[n][n];
        copy(result, a, 0, 0);
        copy(result, b, 0, n / 2);
        copy(result, c, n / 2, 0);
        copy(result, d, n / 2, n / 2);
        return result;
    }

    private static void copy(int[][] result, int[][] a, int m, int n) {
        for (int i = 0; i < a.length; i++) {
            for (int j = 0; j < a.length; j++) {
                result[m + i][n + j] = a[i][j];
            }
        }
    }

用Java加入四个二维数组

2 个答案: