信息学奥赛一本通 1125：矩阵乘法

【题目描述】

计算两个矩阵的乘法。n×m阶的矩阵A乘以m×k阶的矩阵B得到的矩阵C 是n×k阶的，且C[i][j] = A[i][0]×B[0][j] + A[i][1]×B[1][j] + …… +A[i][m-1]×B[m-1][j](C[i][j]表示C矩阵中第i行第j列元素)。

【输入】

第一行为n, m, k，表示A矩阵是n行m列，B矩阵是m行k列，n, m, k均小于100。

然后先后输入A和B两个矩阵，A矩阵n行m列，B矩阵m行k列，矩阵中每个元素的绝对值不会大于1000。

【输出】

输出矩阵C，一共n行，每行k个整数，整数之间以一个空格分开。

【输入样例】

3 2 3
1 1
1 1
1 1
1 1 1
1 1 1

【输出样例】

2 2 2
2 2 2
2 2 2

代码1:

#include<stdio.h>
int main(){
	int n, m, k, i, j, h;
	scanf("%d %d %d",&n, &m, &k);
	int num1[n][m], num2[m][k], num3[n][k];
	for (i=0;i<n;i++){
		for(j=0;j<m;j++){
			scanf("%d",&num1[i][j]);
		}
	}
	for(i=0;i<m;i++){
		for(j=0;j<k;j++){
			scanf("%d",&num2[i][j]);
		}
	}
	for(i=0;i<n;i++){
		for(j=0;j<k;j++){
			num3[i][j]=0;
		}
	}
	for (i=0;i<n;i++){
		for(j=0;j<k;j++){
			for(h=0;h<m;h++){
				num3[i][j] += num1[i][h]*num2[h][j];
			}
		}
	}
	for (i=0;i<n;i++){
		for(j=0;j<k;j++){
			printf("%d ",num3[i][j]);
		}
		printf("\n");
	}
	return 0;
}

代码2：

#include <stdio.h>
void transpose(int m, int n, int source[m][n], int dest[n][m]) {
    int i, j;
    for (i = 0; i < n; i++) {
        for (j = 0; j < m; j++) {
            dest[i][j] = source[j][i];
        }
    }
}
void matrix_multiply(int n, int m, int k, int num1[n][m], int num2_t[k][m], int num3[n][k]) {
    int i, j, h;
    // 初始化 num3 数组
    for (i = 0; i < n; i++) {
        for (j = 0; j < k; j++) {
            num3[i][j] = 0;
        }
    }
    for (i = 0; i < n; i++) {
        for (j = 0; j < k; j++) {
            for (h = 0; h < m; h++) {
                num3[i][j] += num1[i][h] * num2_t[j][h];
            }
        }
    }
}
int main() {
    int n, m, k, i, j;
    scanf("%d %d %d", &n, &m, &k);
    int num1[n][m], num2[m][k], num3[n][k], num2_t[k][m];
    for (i = 0; i < n; i++) {
        for (j = 0; j < m; j++) {
            scanf("%d", &num1[i][j]);
        }
    }
    for (i = 0; i < m; i++) {
        for (j = 0; j < k; j++) {
            scanf("%d", &num2[i][j]);
        }
    }
    transpose(m, k, num2, num2_t);
    matrix_multiply(n, m, k, num1, num2_t, num3);
    for (i = 0; i < n; i++) {
        for (j = 0; j < k; j++) {
            printf("%d ", num3[i][j]);
        }
        printf("\n");
    }
    return 0;
}