C语言实现多维数组内重叠3x3子矩阵平均值计算的问题
实现3x3重叠子矩阵平均值计算的函数改写方案
核心逻辑说明
首先明确输出矩阵的尺寸规则:
- 输入矩阵尺寸为
rows行 × cols列时,所有合法可重叠3x3子矩阵的数量是(rows-2) × (cols-2)个,因此你提前分配的matrixOut需要符合这个尺寸,避免越界。 - 遍历原始矩阵所有可以作为3x3子矩阵左上角的坐标
(i,j),其中i的取值范围是0 ≤ i ≤ rows-3,j的取值范围是0 ≤ j ≤ cols-3 - 对每个左上角坐标,累加其覆盖的3x3区域所有元素值,除以9得到平均值,写入输出矩阵对应坐标
(i,j)的位置
改写后的函数实现
void matavg(float *matrixIn, float *matrixOut, int rows, int cols, float *globalAvg) { // 保留原有的全局平均值计算逻辑(不需要可直接删除) float totalSum = 0; for (int i = 0; i < rows; i++) { for (int j = 0; j < cols; j++) { totalSum += *(matrixIn + i * cols + j); } } *globalAvg = totalSum / (rows * cols); // 新增3x3子矩阵平均值计算逻辑 int outRows = rows - 2; int outCols = cols - 2; // 遍历所有3x3子矩阵的左上角坐标 for (int i = 0; i < outRows; i++) { for (int j = 0; j < outCols; j++) { float subSum = 0; // 累加当前3x3子矩阵的所有元素 for (int di = 0; di < 3; di++) { for (int dj = 0; dj < 3; dj++) { subSum += *(matrixIn + (i + di) * cols + (j + dj)); } } // 计算平均值写入输出矩阵 *(matrixOut + i * outCols + j) = subSum / 9.0f; } } }
调用注意事项
- 调用前请确保
matrixOut已经分配了至少(rows-2)*(cols-2)个float类型的存储空间,避免内存越界 - 如果输入矩阵行数/列数小于3,函数不会生成任何输出结果,调用前可以先做合法性判断
- 如果不需要全局平均值计算,可以删除对应的代码段,同时去掉函数的
globalAvg参数即可
内容的提问来源于stack exchange,提问作者ridgiz
相关产品推荐
相关产品推荐

