nyoj A*B Problem II 623 (矩阵相乘)

A*B Problem II

时间限制： 1000 ms  |           内存限制： 65535 KB

难度： 1

描述

ACM的C++同学有好多作业要做，最头痛莫过于线性代数了，因为每次做到矩阵相乘的时候，大量的乘法都会把他搞乱，所以他想请你写个程序帮他检验一下计算结果是否正确。

输入

有多组测试数据，每行给出一组m,n,k（0<m,n,k<=50）。m,n,k表示两个矩阵的大小,其中：
矩阵A：m行n列。
矩阵B：n行k列。
接下来给出m*n个数表示矩阵A和n*k个数表示矩阵B，对于每个数s，0<=s<=1000。
当m，n，k同时为0时结束。

输出

计算两个矩阵的乘积并输出。

样例输入

2 1 3
1
2
1 2 3
2 2 3
1 2
3 4
1 0 1
0 1 0
0 0 0

样例输出

1 2 3
2 4 6
1 2 1
3 4 3

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define INF 0x3f3f3f3f
#define ll long long
#define N 60
using namespace std;
struct mat
{
	ll m[N][N];
}a,b;
int m,n,kk;
mat multi(mat a,mat b)
{
	int i,j,k;
	mat c;
	for(i=0;i<m;i++)
	{
		for(j=0;j<kk;j++)
		{
			c.m[i][j]=0;
			for(k=0;k<n;k++)
			{
				c.m[i][j]+=a.m[i][k]*b.m[k][j];
			}
		}
	}
	return c;
}
int main()
{
	
	int i,j;
	while(scanf("%d%d%d",&m,&n,&kk),m|n|kk)
	{
		for(i=0;i<m;i++)
			for(j=0;j<n;j++)
				scanf("%d",&a.m[i][j]);
		for(i=0;i<n;i++)
			for(j=0;j<kk;j++)
				scanf("%d",&b.m[i][j]);
		mat ans=multi(a,b);
		for(i=0;i<m;i++)
		{
			for(j=0;j<kk;j++)
			{
				printf(j?" %d":"%d",ans.m[i][j]);
			}
			printf("\n");
		}
	}
	return 0;
}