基于三元组顺序表实现稀疏矩阵的加法(C++)

假设稀疏矩阵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;	
}

结果如下

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值