题目描述
计算两个矩阵的乘法。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。
输入
第一行为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
思路:根据上述矩阵乘法的定义,n*m和m*k的两个矩阵相乘得到n*k的矩阵,而根据表达式,我们可以使用三个循环来完成矩阵乘法运算。
代码如下:
#include <iostream>
#include <cstring>
#define N 100
using namespace std;
int main()
{
int n,m,k;
cin>>n>>m>>k;
int num1[N][N],num2[N][N],operate[N][N];
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
cin>>num1[i][j];
for(int i=0;i<m;i++)
for(int j=0;j<k;j++)
cin>>num2[i][j];
memset(operate,0,sizeof(operate));
for(int i=0;i<n;i++)
for(int j=0;j<k;j++)
for(int l=0;l<m;l++)
{
operate[i][j] +=num1[i][l]*num2[l][j];
}
for(int i=0;i<n;i++)
{
for(int j=0;j<k;j++)
{
cout<<operate[i][j]<<" ";
}
cout<<endl;
}
}