矩阵乘法C++实现

原创已于 2023-10-09 00:49:18 修改 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

于 2022-04-01 18:35:32 首次发布

C++ 专栏收录该内容

16 篇文章

订阅专栏

这段代码实现了两个二维整数矩阵的乘法操作。首先获取输入矩阵的尺寸，然后通过三重循环进行矩阵乘法，将结果存储在新的矩阵C中。最后返回矩阵C。

在这里插入图片描述

std::vector<std::vector<int>> matrix_mutiplication(const std::vector<std::vector<int>>& A,const std::vector<std::vector<int>>& B)
{
	int A_row = A.size(), A_col = A[0].size(), B_row = B.size(), B_col = B[0].size();
	assert(A_col == B_row);

	std::vector<std::vector<int>> C(A_row, std::vector<int>(B_col));
	for (int i = 0; i < C.size(); ++i)
	{
		for (int j = 0; j < C[0].size(); ++j)
		{
			for (int k = 0; k < A_col; ++k)
				C[i][j] += A[i][k] * B[k][j]; //C[i][j]等于A的第i行乘B的第j列
			//std::cout << C[i][j] << " ";
		}
		//std::cout << endl;
	}

	return C;
}