矩阵乘法

转自：

https://blog.youkuaiyun.com/cxc19890214/article/details/45363267/

https://blog.youkuaiyun.com/yang20141109/article/details/51024373

方法一：数组实现：

 

1. #include<iostream>  
2. using namespace std;  
3. #define SIZE_M 2  
4. #define SIZE_N 3  
5. #define SIZE_S 4  
6.   
7. int main(){  
8.   
9.     int matrix_a[SIZE_M] [SIZE_N]={{1,2,3},{4,5,6}};//a矩阵2X3  
10.     int matrix_b[SIZE_N] [SIZE_S]={{1,2,3,4},{5,6,7,8},{1,2,3,4}};//b矩阵3X4  
11.     int matrix_result[SIZE_M][SIZE_S];//结果矩阵2X4  
12.       
13.   
14.     for(int m=0;m<SIZE_M;m++){  
15.         for(int s=0;s<SIZE_S;s++){  
16.             matrix_result[m][s]=0;//变量使用前记得初始化,否则结果具有不确定性  
17.             for(int n=0;n<SIZE_N;n++){  
18.                 matrix_result[m][s]+=matrix_a[m][n]*matrix_b[n][s];  
19.             }  
20.         }  
21.     }  
22.     for(int m=0;m<SIZE_M;m++){  
23.         for(int s=0;s<SIZE_S;s++){  
24.             cout<<matrix_result[m][s]<<"\t";  
25.         }  
26.         cout<<endl;  
27.     }  
28.   
29.     return 0;  
30.   
31. }  

方法二：vector实现

 

1. #include<iostream>  
2. #include<vector>  
3. using namespace std;  
4.   
5. vector<vector<int>> matrix_multiply(vector<vector<int>> arrA, vector<vector<int>> arrB)  
6. {  
7.     //矩阵arrA的行数  
8.     int rowA = arrA.size();  
9.     //矩阵arrA的列数  
10.     int colA = arrA[0].size();  
11.     //矩阵arrB的行数  
12.     int rowB = arrB.size();  
13.     //矩阵arrB的列数  
14.     int colB = arrB[0].size();  
15.     //相乘后的结果矩阵  
16.     vector<vector<int>>  res;  
17.     if (colA != rowB)//如果矩阵arrA的列数不等于矩阵arrB的行数。则返回空  
18.     {  
19.         return res;  
20.     }  
21.     else  
22.     {  
23.         //设置结果矩阵的大小，初始化为为0  
24.         res.resize(rowA);  
25.         for (int i = 0; i < rowA; ++i)  
26.         {  
27.             res[i].resize(colB);  
28.         }  
29.   
30.         //矩阵相乘  
31.         for (int i = 0; i < rowA; ++i)  
32.         {  
33.             for (int j = 0; j < colB; ++j)  
34.             {  
35.                 for (int k = 0; k < colA; ++k)  
36.                 {  
37.                     res[i][j] += arrA[i][k] * arrB[k][j];  
38.                 }  
39.             }  
40.         }  
41.     }  
42.     return res;  
43. }  
44.   
45. int main(void)  
46. {  
47.     vector<vector<int>> arrA = { { 2, 1 }, { 4, 3 } };  
48.     vector<vector<int>> arrB = { { 1, 2 }, { 1, 0 } };  
49.     //vector<vector<int>> arrA = { { 1, 2, 3 }, { 4, 5, 6 } };  
50.     //vector<vector<int>> arrB = { { 1, 4 }, { 2, 5 }, { 3, 6 } };  
51.     vector<vector<int>> res = matrix_multiply(arrA, arrB);  
52.     system("pause");  
53.     return 0;  
54. }