行逻辑链接的矩阵乘法

  1 #include <stdio.h>  
  2 #include <string.h>  
  3 #include <iostream>  
  4 #include <string>  
  5 #include <math.h>  
  6 #include <algorithm>  
  7 #include <vector>  
  8 #include <stack>  
  9 #include <queue>  
 10 #include <set>  
 11 #include <map>
 12 const int INF=0x3f3f3f3f;  
 13 typedef long long LL;
 14 const int mod=1e9+7;
 15 const int maxn=1e4+2510;  
 16 using namespace std;  
 17   
 18 typedef struct  
 19 {  
 20     int row;//行
 21     int col;//列
 22     int val;//值
 23 }Triple;  
 24   
 25 typedef struct  
 26 {  
 27     Triple data[maxn];//三元组
 28     int rpos[maxn];//每一行的第一个非零元在三元组表中的位置
 29     int rows;//矩阵的总行数
 30     int cols;//矩阵的总列数
 31     int nums;//矩阵的非零元素个数
 32 }Matrix;
 33 
 34 void MulMatrix(Matrix A,Matrix B,Matrix *C)
 35 {
 36     if(A.cols!=B.rows)//如果矩阵A的列数与矩阵B的行数不等，则不能做矩阵乘运算
 37         return ;
 38     C->rows=A.rows;
 39     C->cols=B.cols;
 40     C->nums=0;
 41     if(A.nums*B.nums==0)//如果其中任意矩阵的元素个数为零，做乘法元素没有意义，全是0
 42         return ;
 43     int ccol;
 44     for(int arow=1;arow<=A.rows;arow++)
 45     {
 46         int temp[maxn]={0};//创建一个临时存储乘积结果的数组，且初始化为0，遍历每次都需要清空
 47         C->rpos[arow]=C->nums+1;
 48         int tp;
 49         if(arow<A.rows)
 50             tp=A.rpos[arow+1];//获取矩阵A的下一行第一个非零元素在data数组中位置
 51         else
 52             tp=A.nums+1;//若当前行是最后一行，则取最后一个元素+1
 53         //遍历当前行的所有的非0元素
 54         for(int p=A.rpos[arow];p<tp;p++)
 55         {
 56             int brow=A.data[p].col;//取该非0元素的列数，便于去B中找对应的做乘积的非0元素
 57             int t;
 58             // 判断如果对于A中非0元素，找到矩阵B中做乘法的那一行中的所有的非0元素
 59             if(brow<B.rows)
 60                 t=B.rpos[brow+1];
 61             else
 62                 t=B.nums+1;
 63             //遍历找到的对应的非0元素，开始做乘积运算
 64             for(int q=B.rpos[brow];q<t;q++)
 65             {
 66                 //得到的乘积结果，每次和temp数组中相应位置的数值做加和运算
 67                 ccol=B.data[q].col;
 68                 temp[ccol]+=A.data[p].val*B.data[q].val;
 69             }
 70         }
 71         //矩阵C的行数等于矩阵A的行数，列数等于矩阵B的列数，所以，得到的temp存储的结果，也会在C的列数的范围内
 72         for(ccol=1;ccol<=C->cols;ccol++)
 73         {
 74             //由于结果可以是0，而0不需要存储，所以在这里需要判断
 75             if(temp[ccol])
 76             {
 77                 C->nums++;
 78                 if(C->nums > maxn)
 79                     return;
 80                 C->data[C->nums].val=temp[ccol];
 81                 C->data[C->nums].row=arow;
 82                 C->data[C->nums].col=ccol;
 83             }
 84         }
 85     }
 86 }
 87 
 88 int main()  
 89 {
 90     Matrix A,B,C;
 91     scanf("%d %d",&A.rows,&A.cols);
 92     A.nums=0;
 93     for(int i=1;i<=A.rows;i++)
 94     {
 95         A.rpos[i]=A.nums+1;
 96         for(int j=1;j<=A.cols;j++)
 97         {
 98             int x;
 99             scanf("%d",&x);
100             if(x)
101             {
102                 A.nums++;  
103                 A.data[A.nums].row=i;
104                 A.data[A.nums].col=j;
105                 A.data[A.nums].val=x;
106             }
107         }
108     }
109     scanf("%d %d",&B.rows,&B.cols);  
110     B.nums=0;
111     for(int i=1;i<=B.rows;i++)  
112     { 
113         B.rpos[i]=B.nums+1;
114         for(int j=1;j<=B.cols;j++)  
115         {  
116             int x;  
117             scanf("%d",&x);  
118             if(x)  
119             {  
120                 B.nums++;  
121                 B.data[B.nums].row=i;  
122                 B.data[B.nums].col=j;  
123                 B.data[B.nums].val=x;  
124             }
125         }
126     }
127     MulMatrix(A,B,&C); 
128     int cnt=1; 
129     for(int i=1;i<=C.rows;i++)//输出转置后的矩阵   
130     {
131         for(int j=1;j<=C.cols;j++)
132         {
133             int a,b;
134             a=C.data[cnt].row;
135             b=C.data[cnt].col;
136             if(a==i&&b==j)
137             {
138                 printf("%d ",C.data[cnt].val);
139                 cnt++;
140             }
141             else
142                 printf("%d ",0);
143         }
144         printf("\n");
145     }
146     return 0;
147 }