华为机试题---矩阵乘法
题目描述
如果A是个x行y列的矩阵,B是个y行z列的矩阵,把A和B相乘,其结果将是另一个x行z列的矩阵C。这个矩阵的每个元素是由下面的公式决定的
输入描述:
输入包含多组数据,每组数据包含:
第一行包含一个正整数x,代表第一个矩阵的行数
第二行包含一个正整数y,代表第一个矩阵的列数和第二个矩阵的行数
第三行包含一个正整数z,代表第二个矩阵的列数
之后x行,每行y个整数,代表第一个矩阵的值
之后y行,每行z个整数,代表第二个矩阵的值
输出描述:
对于每组输入数据,输出x行,每行z个整数,代表两个矩阵相乘的结果
示例1
输入
2
3
2
1 2 3
3 2 1
1 2
2 1
3 3
输出
14 13
10 11
#include <iostream>
#include <vector>
using namespace std;
int main()
{
int x, y, z;
while (cin >> x >> y >> z)
{
vector<vector<int>> v1;
vector<vector<int>> v2;
for (int i = 0; i < x; i++)
{
vector<int> vTemp;
for (int j = 0; j < y; j++)
{
int t = 0;
cin >> t;
vTemp.push_back(t);
}
v1.push_back(vTemp);
}
for (int i = 0; i < y; i++)
{
vector<int> vTemp;
for (int j = 0; j < z; j++)
{
int t = 0;
cin >> t;
vTemp.push_back(t);
}
v2.push_back(vTemp);
}
int retBuf[100][100] = { 0 };
for (int i = 0; i < x; i++)
{
for (int j = 0; j < z; j++)
{
for (int k = 0; k < y; k++)
{
retBuf[i][j] += v1[i][k] * v2[k][j];
}
}
}
for (int i = 0; i < x; i++)
{
for (int j = 0; j < z; j++)
{
cout << retBuf[i][j] << " ";
}
cout << endl;
}
}
return 0;
}