#include<iostream>
#include<vector>
using namespace std;
vector<vector<int>> matrix_multiply(vector<vector<int>> arrA, vector<vector<int>> arrB)
{
//矩阵arrA的行数
int rowA = arrA.size();
//矩阵arrA的列数
int colA = arrA[0].size();
//矩阵arrB的行数
int rowB = arrB.size();
//矩阵arrB的列数
int colB = arrB[0].size();
//相乘后的结果矩阵
vector<vector<int>> res;
if (colA != rowB)//如果矩阵arrA的列数不等于矩阵arrB的行数。则返回空
{
return res;
}
else
{
//设置结果矩阵的大小,初始化为为0
res.resize(rowA);
for (int i = 0; i < rowA; ++i)
{
res[i].resize(colB);
}
//矩阵相乘
for (int i = 0; i < rowA; ++i)
{
for (int j = 0; j < colB; ++j)
{
for (int k = 0; k < colA; ++k)
{
res[i][j] += arrA[i][k] * arrB[k][j];
}
}
}
}
return res;
}
int main(void)
{
vector<vector<int>> arrA = { { 1, 0, 0 },{ 0, 1, 0 },{ 0, 0, 1 } };
vector<vector<int>> arrB = { { 1, 2, 3 },{ 4, 5, 6 },{ 7, 8, 9 } };
//vector<vector<int>> arrA = { { 1, 2, 3 }, { 4, 5, 6 } };
//vector<vector<int>> arrB = { { 1, 4 }, { 2, 5 }, { 3, 6 } };
vector<vector<int>> res = matrix_multiply(arrA, arrB);
system("pause");
return 0;
}