矩阵相乘 C++

#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;
}