ZJYYC1199. 矩阵乘法
题目描述
矩阵乘法运算:A(n × m)的每一行中的数字对应乘以B(m×k)的每一列的数字把结果相加起来。相加的前提是:A矩阵的列的值等于B矩阵的行的值。
设A为 n × m 的矩阵,B为 m × k 的矩阵,那么称 的矩阵C为矩阵A与B的乘积,记作 C = AB,其中矩阵C(n × k)中的第 i 行第 j 列元素可以表示为:
如图所示:
输入描述
测试样例由多组测试数据组成。
每个样例包含若干行,第一行输入三个整数 n、m、k;
接下来 n 行,每行输入 m 个整数a i j ,表示矩阵A;
再接下来 m 行,每行输入 k 个整数a i j ,表示矩阵B;
1<=n,m;k<=200,0<=a(i)<=1000
输出描述
对于每个样例,输出乘法运算之后的矩阵。每个样例之后输出一个空行。
测试样例1
输入
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 <iostream>
#include <vector>
using namespace std;
int main() {
int n, m, k;
while (cin >> n >> m >> k) {
vector<vector<int>> matrixA(n, vector<int>(m));
vector<vector<int>> matrixB(m, vector<int>(k));
vector<vector<int>> result(n, vector<int>(k, 0));
for (int i = 0; i < n; ++i) {
for (int j = 0; j < m; ++j) {
cin >> matrixA[i][j];
}
}
for (int i = 0; i < m; ++i) {
for (int j = 0; j < k; ++j) {
cin >> matrixB[i][j];
}
}
for (int i = 0; i < n; ++i) {
for (int j = 0; j < k; ++j) {
for (int l = 0; l < m; ++l) {
result[i][j] += matrixA[i][l] * matrixB[l][j];
}
}
}
for (int i = 0; i < n; ++i) {
for (int j = 0; j < k; ++j) {
cout << result[i][j] << (j < k - 1 ? ' ' : '\n');//这样的目的是为了避免格式错误
}
}
cout<<endl;
}
return 0;
}