1127: 矩阵乘积

题目描述

计算两个矩阵A和B的乘积。

输入

第一行三个正整数m、p和n，0<=m,n,p<=10，表示矩阵A是m行p列，矩阵B是p行n列；

接下来的m行是矩阵A的内容，每行p个整数，用空格隔开；

最后的p行是矩阵B的内容，每行n个整数，用空格隔开。

输出

输出乘积矩阵：输出占m行，每行n个数据，以空格隔开。

样例输入

2 3 4

1 0 1
0 0 1

1 1 1 3
4 5 6 7
8 9 1 0

样例输出

9 10 2 3
8 9 1 0

代码

#include<stdio.h>
#include<stdlib.h>
int main(){
	int m,p,n;
	scanf("%d %d %d",&m, &p,&n);
	int *matrix1 = (int *)malloc(sizeof(int)*(m*p));
	int *matrix2 = (int *)malloc(sizeof(int)*(p*n));
	for(int i=0;i<m;i++){
		for(int j=0;j<p;j++){
			scanf("%d", matrix1+i*p+j);
		}
	}
	for(int i=0;i<p;i++){
		for(int j=0;j<n;j++){
			scanf("%d", matrix2+i*n+j);
		}
	}
	for(int i=0;i<m;i++){
		for(int j=0;j<n;j++){
			int sum = 0;
			for(int k=0; k<p; k++){
				sum+=matrix1[i*p+k]*matrix2[k*n+j];
			}
			printf("%d ", sum);
		}
		printf("\n");
	}
	return 0;
}

提交结果截图