C++ 实现矩阵乘法

矩阵相乘的时间复杂度为o(n*n*n)。使用vector<vector<int>>来表示矩阵。

#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)
{
	v