C++ 矩阵乘法函数实现

C++ 矩阵乘法函数实现

我们可以使用给定的函数原型 int** multiplication(int a[][1], int aM, int aN, int b[][1], int bM, int bN) 实现矩阵乘法。下面是具体实现的代码：

int** multiplication(int a[][1], int aM, int aN, int b[][1], int bM, int bN){

    // 首先判断两个矩阵是否满足相乘的条件
    if(aN != bM){
        return NULL;
    }

    // 创建结果矩阵
    int **result = new int*[aM];
    for(int i=0; i<aM; i++){
        result[i] = new int[bN];
    }

    // 计算矩阵乘法
    for(int i=0; i<aM; i++){
        for(int j=0; j<bN; j++){
            result[i][j] = 0;
            for(int k=0; k<aN; k++){
                result[i][j] += a[i][k] * b[k][j];
            }
        }
    }

    return result;
}

这个函数接受两个二维数组，以及它们的行数和列数作为参数。它首先检查两个矩阵是否满足相乘的条件，即第一个矩阵的列数必须等于第二个矩阵的行数。然后创建一个结果矩阵，其行数等于第一个矩阵的行数，列数等于第二个矩阵的列数。接下来，它使用三重嵌套的循环来计算矩阵乘法，其中外层循环遍历结果矩阵的行，中间循环遍历结果矩阵的列，内层循环遍历第一个矩阵的列（或第二个矩阵的行）。最后，它返回结果矩阵。