假设稀疏矩阵A和B均以三元组顺序表作为存储结构,另设三元组表C存放结果矩阵。
#include <iostream>
using namespace std;
const int MAXSIZE=100;
typedef double ElemType;
typedef struct{
int i,j;
ElemType e;
}Triple;
typedef struct{
Triple data[MAXSIZE+1];
int mu,nu,tu;
}TSMatrix;
const int MAXMN=100;
//检查相加是否合法
bool AddTSM(TSMatrix A,TSMatrix B,TSMatrix &C)
{
if(A.mu!=B.mu||A.nu!=B.nu)
{
cout<<"两矩阵不具有相同的行(列)数"<<endl;
return false;
}
//两稀疏矩阵相加
C.mu=A.mu;
C.nu=A.nu;
C.tu=0;
int k1=1,k2=1;
while(k1<=A.tu&&k2<=B.tu)
{
if(A.data[k1].i < B.data[k2].i)
{
//将A的元素赋给C
C.data[++C.tu].e=A.data[k1].e;
C.data[C.tu].i=A.data[k1].i;
C.data[C.tu].j=A.data[k1].j;
k1++;
}
else if(A.data[k1].i==B.data[k2].i)
{
if(A.data[k1].j<B.data[k2].j)
{
C.data[++C.tu].e=A.data[k1].e;
C.data[C.tu].i=A.data[k1].i;
C.data[C.tu].j=A.data[k1].j;
k1++;
}
else if(A.data[k1].j==B.data[k2].j)
{
ElemType add;
if((add=A.data[k1].e+B.data[k2].e)!=0)
{
C.data[++C.tu].e=add;
C.data[C.tu].i=A.data[k1].i;
C.data[C.tu].j=A.data[k1].j;
}
k1++;k2++;
}
else
{
C.data[++C.tu].e=B.data[k2].e;
C.data[C.tu].i=B.data[k2].i;
C.data[C.tu].j=B.data[k2].j;
k2++;
}
}
}
if(k1==A.tu+1)
{
for(;k2<=B.tu;k2++)
{
C.data[++C.tu].e=B.data[k2].e;
C.data[C.tu].i=B.data[k2].i;
C.data[C.tu].j=B.data[k2].j;
}
}
else if(k2==B.tu+1)
{
for(;k1<=A.tu;k1++)
{
C.data[++C.tu].e=A.data[k1].e;
C.data[C.tu].i=A.data[k1].i;
C.data[C.tu].j=A.data[k1].j;
}
}
return true;
}
void TraverseTSM(TSMatrix M)
{
for(int i=1;i<=M.tu;i++)
cout<<M.data[i].i<<'\t'<<M.data[i].j<<'\t'<<M.data[i].e<<endl;
}
int main()
{
TSMatrix A,B,C,TA;
C.mu=1;C.nu=1;C.tu=0;
A.mu=6;A.nu=7;A.tu=8;
A.data[1].i=1;A.data[1].j=2;A.data[1].e=12;
A.data[2].i=1;A.data[2].j=3;A.data[2].e=9;
A.data[3].i=3;A.data[3].j=1;A.data[3].e=-3;
A.data[4].i=3;A.data[4].j=6;A.data[4].e=14;
A.data[5].i=4;A.data[5].j=3;A.data[5].e=24;
A.data[6].i=5;A.data[6].j=2;A.data[6].e=-18;
A.data[7].i=6;A.data[7].j=1;A.data[7].e=15;
A.data[8].i=6;A.data[8].j=4;A.data[8].e=-1;
B.mu=6;B.nu=7;B.tu=8;
B.data[1].i=1;B.data[1].j=2;B.data[1].e=12;
B.data[2].i=1;B.data[2].j=3;B.data[2].e=9;
B.data[3].i=3;B.data[3].j=1;B.data[3].e=-3;
B.data[4].i=3;B.data[4].j=6;B.data[4].e=14;
B.data[5].i=4;B.data[5].j=3;B.data[5].e=24;
B.data[6].i=5;B.data[6].j=2;B.data[6].e=-18;
B.data[7].i=6;B.data[7].j=1;B.data[7].e=15;
B.data[8].i=6;B.data[8].j=4;B.data[8].e=-1;
cout<<"A="<<endl;
TraverseTSM(A);
cout<<"B="<<endl;
TraverseTSM(B);
cout<<"C=A+B"<<endl;
AddTSM(A,B,C);
TraverseTSM(C);
return 0;
}
结果如下