NOI / 1.8编程基础之多维数组题 09:矩阵乘法

 

09:矩阵乘法

• 查看
• 提交
• 统计
• 提问

总时间限制: 

1000ms

内存限制: 

65536kB

描述

计算两个矩阵的乘法。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

#include<stdio.h>
int main()
{
	int m,n,k;
	scanf("%d%d%d",&n,&m,&k);
	int a[100][100],b[100][100],c[100][100];
	int i,j,x;
	for(i=0;i<n;i++)
	for(j=0;j<m;j++){
		scanf("%d",&a[i][j]);
	} 
	
	for(i=0;i<m;i++)
	for(j=0;j<k;j++){
		scanf("%d",&b[i][j]);
	} 
	
	for(i=0;i<n;i++)
	for(j=0;j<k;j++){
		c[i][j]=0;
		for(x=0;x<m;x++)
		c[i][j]+=a[i][x]*b[x][j];
	}

	
	for(i=0;i<n;i++){
		for(j=0;j<k;j++){
			printf("%d",c[i][j]);
			printf(" ");
		}
		printf("\n");
		}
	return 0;
}

如何用编程语言实现矩阵乘法？

设A为m×n的矩阵，B为n×t的矩阵，那么称m×t的矩阵C为矩阵A与矩阵B的乘积，记为C=AB，其中矩阵C的第i行第j列的元素可以表示为：

图源 百度百科：矩阵乘法

示例：

根据矩阵乘法规则，每得到矩阵C的一个元素，需要将n次乘法的结果相加。

示例中14=2×1+3×4（共两次乘法）。

乘积结果C每个元素的时间复杂度为O(n),元素个数为m×t。

因此，总的时间复杂度为O(m×n×t),即为O( n^3 )级。