java - 尝试使用第三个矩阵同时初始化两个矩阵 Row 和 Col

Question

矩阵行是一行中所有数字的总和，矩阵列是一行中所有数字的sm如果我们有这个矩阵 m:

[1 2 3 4]
[3 4 5 1]
[1 2 0 1]

我们的矩阵行是:

[1 3 6 10]
[3 7 12 13]
[1 3 3 4]

我们的矩阵列是:

[1 2 3 4]
[4 6 8 5]
[5 8 8 6]

我正在编写一种同时填充两个 NxM 矩阵的方法。矩阵行应按其行填充，矩阵列应按其列填充。

这是我的代码:

Row[0][0] = m[0][0]
Col[0][0] = m[0][0]
for(int i = 0; i < nrRow; i++){
  for(int j = 0; i < nrRow; j++){
     Row[i][j] = Row[i][j]+m[i][j]
  }
}

但我不知道如何填写 Col 矩阵。整个操作应该在 O(n^2) 内完成。

Answer 1

1. 对于colMatrix，对于每个索引[i, j]，您必须用m[i][j] + 填充它colMatrix[i][j - 1]。其中 m 是原始矩阵，colMatrix 是您要填充的新矩阵。现在，您必须单独考虑 j == 0 的情况，在这种情况下，您将需要 colMatrix[i][j] = m[i][j]

2. 与 rowMatrix 类似，对于每个索引 [i, j]，其值将为 m[i][j] + rowMatrix[ i - 1][j]。对于 i == 0，您将得到 rowMatrix[i][j] = m[i][j]。

代码如下:

    int[][] m = {{1, 2, 3, 4}, {3, 4, 5, 1}, {1, 2, 0, 1}};

    int[][] rowMatrix = new int[m.length][m[0].length];
    int[][] colMatrix = new int[m.length][m[0].length];

    for (int i = 0; i < m.length; i++) {
        for (int j = 0; j < m[0].length; j++) {

            if (j == 0) rowMatrix[i][j] = m[i][j]; 
            else        rowMatrix[i][j] = m[i][j] + rowMatrix[i][j - 1];

            if (i == 0) colMatrix[i][j] = m[i][j];
            else        colMatrix[i][j] = m[i][j] + colMatrix[i - 1][j];

        }
    }