正如您所说,您编写的代码(两个 for 循环都停止在 3)使您可以在从 [0][0] 开始的 3x3 子矩阵中计算您想要的数字。你现在需要做的是:
确定您拥有的所有起点,确保
从那里开始有 3 行 3 列,包括
您所在的元素,并在每个可能的情况下调用该函数
起点和
扩展该代码,使其可以从
矩阵。
下面是我们如何继续第 1 点:
您有一个 4x4 矩阵,但我们可以轻松地将其推广到 NxM 矩阵。 N表示有N行,即每列有N个元素。所以让我们考虑一个列。如果你有 N 个元素,你能找到多少组 3 个连续的元素?当心,它不是 N-3(这看起来是一个显而易见的答案,但它是错误的):
- N=3 个元素只有 1 个(如果有 123 个,则该组为 123 个)
- N=4 个元素有 2 个(如果有 1234 个,则组为 123 和 234)
- N=5 个元素有 3 个(如果有 12345,则组为 123,
234, 345)
所以一般来说你有 N - 3 + 1 个组(当然你可以认为它是 N-2,但我更喜欢用 3 来表示,因为这是我们每次取的元素数量)。
同样的方法可以应用于列:如果它们是 M,你可以采取 M - 3 + 1 个组。
也就是说,您可以在 NxM 矩阵中找到多少个 3x3 子矩阵?首先,您可以找到所有 3 的“垂直”组,即 N-3+1。对于他们每个人,你能组成多少个 3 人的“水平”组? M-3+1。那么你总共有 (N-3+1)*(M-3+1) 个 3x3 子矩阵。这是检查每个子矩阵必须进行的迭代次数。这些 N-3+1 个元素中的每一个都是一行(我们正在垂直移动,即沿着一列,所以这些是行),所以这是一个我们可以开始检查矩阵的元素 - 只要我们有足够的水平空间,即有足够的列。对于这些行中的每一行,我们必须检查 M-3+1 列(沿行水平移动),其中可以开始一个 3x3 子矩阵。
int min_found = 0; // So far we haven't found a minimum yet
int min_value, min_row, min_column;
int max_rows = (N-3+1);
int max_columns = (M-3+1);
for (int start_row = 0; start_row < max_rows; start_row++) {
for (int start_column = 0; start_column < max_columns; start_column++) {
int current_sum = calculate_sum_of_submatrix(start_row, start_column);
if ( (min_found == 0) || (min_found == 1 && current_sum < min_value) )
min_value = current_sum;
min_row = start_row;
min_column = start_column;
min_found = 1;
}
}
}
cout << "The 3x3 submatrix whose elements have the minimum sum "
<< "starts at row " << min_row << " and column " << min_column
<< ", and the sum is " << min_sum << endl;
在您的情况下(N=4,M=4),我们需要检查 N-3+1=2 行和 M-3+1=2 列。所以我们从 [0][0] 开始,然后内部循环将我们带到 [0][1],然后内部循环将结束,我们将继续到外部循环,这将带我们到第 1 行,内部循环将检查 [1][0] 和 [1][1]。然后就结束了。到这个时候,我们将检查所有 4 个 3x3 子矩阵,并且我们会知道最小值。
现在,第 2 点:我们必须调整您已经拥有的功能,以便从我们想要的位置开始,而不是总是从 [0][0] 开始。这很容易,我们只需在由起始位置(在您的代码中始终为 [0][0])和您已经拥有的 [i][j] 索引的总和给出的位置读取矩阵.像这样:
int calculate_sum_of_submatrix(int start_row, int start_column) {
int sum = 0;
for (int i = 0; i < 3; i++) { // I would replace i with "submatrix_row"
for (int j = 0; j < 3; j++) { // I would replace j with "submatrix_column"
sum += somearray[start_row + i][start_column + j];
}
}
return sum;
}
应该可以的。