C++矩阵相乘

转自:

https://blog.csdn.net/cxc19890214/article/details/45363267/

https://blog.csdn.net/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. }  

猜你喜欢

转载自blog.csdn.net/lyq_csdn/article/details/80633376