CCF NOI1050. 矩阵乘法

时间限制: 1000 ms  空间限制: 262144 KB  具体限制  

题目描述

计算两个矩阵的乘法。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列，矩阵中每个元素的绝对值不会大于100。

输出

输出矩阵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]时可以枚举k=0~m-1，计算的和。

样例输入

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

样例输出

14 10
16 13

 

题记：

用数组模拟，不过c[n][k]我没有存起来再输出，计算完直接输出了，节省内存。

模拟过程见代码吧……实在是不好解释

如果你想知道为什么这样计算，可以参照我在CCF NOI1049. 旋转图像中写的方法试一试。

 

C++程序如下：

#include <iostream>
#include <cstring>

using namespace std;

int main(void)
{
	int n, m, k, l;
	
	cin >> n >> m >> k;
	int a[n][m], b[m][k], c[n][k];
	
	for(int i=0; i<n; i++)
	    for(int j=0; j<m; j++)
	        cin >> a[i][j];
	for(int i=0; i<m; i++)
	    for(int j=0; j<k; j++)
	        cin >> b[i][j];
	
	for(int i=0; i<n; i++){
		for(int j=0; j<k; j++){
			int sum = 0;
			for(l=0; l<m; l++)
			    sum += a[i][l] * b[l][j];
			cout << sum << " ";
		}
		cout << endl;
	}
	   
    return 0;
}