实现两个N*N矩阵相乘的算法

我在人生的道路上迷路了 2019-03-05 10:30:08

实现两个N*N矩阵相乘的算法

...全文

1113 1 打赏收藏转发到动态举报

写回复

用AI写文章

1 条回复

切换为时间正序

请发表友善的回复…

发表回复

qq_43337728 2019-03-30

打赏
举报

#include<iostream> #include<vector> using namespace std; vector<vector<int>> matrix_multiply(vector<vector<int>> arrA, vector<vector<int>> arrB) { int rowA = arrA.size(); int colA = arrA[0].size(); int rowB = arrB.size(); int colB = arrB[0].size(); vector<vector<int>> res; if (colA != rowB) { return res; } else { 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 = { { 2, 1 }, { 4, 3 } }; vector<vector<int>> arrB = { { 1, 2 }, { 1, 0 } }; //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; }