稀疏矩阵--以三元组表为存储结构实现矩阵的相加

 

 

 

 

#include <iostream>
#include <stdio.h>
#include<stdlib.h>
using namespace std;

typedef struct Triple{
	int i,j; //行号，列号 
	int num;//数值 
}Triple;

typedef struct TS{
	Triple data[200]; 
	int mu,nu,tu;//矩阵的行数，列数，非零元素个数 
}TS;

int compare(int r1,int c1,int r2,int c2){
	if(r1>r2)  return (1);
	if(r1<r2) return (-1);
	if(r1==r2&&c1<c2)  return (-1);
	if(r1==r2&&c1>c2) return (1);
	return 0;
}//1在2的后面返回1，相等返回0

void ADDPtoQ(TS *P,int p,TS *Q,int q){
	Q->data[q].i=P->data[p].i;
		      Q->data[q].j=P->data[p].j;
		      Q->data[q].num=P->data[p].num;
		      p=p+1;
		      q=q+1;
}    //在目标矩阵中加入节点 

int main(){
	int m,n,t1,t2;
	int pa,pb,pc;
	pa=pb=pc=0;
	TS *A,*B,*C; 
	A=(TS*)malloc(sizeof(TS));
	B=(TS*)malloc(sizeof(TS));
	C=(TS*)malloc(sizeof(TS));
	A->tu=t1;
	m=100;
	n=100;
	A->mu=B->mu=C->mu=m;
	A->nu=B->nu=C->nu=n;
	C->tu=0; 
	B->tu=t2;   //初始化三个矩阵的行数，列数以及非零元素个数 
	cin>>t1>>t2;	
	for(int p=0;p<t1;p++)
		cin>>A->data[p].i>>A->data[p].j>>A->data[p].num;
	for(int p=0;p<t2;p++)
		cin>>B->data[p].i>>B->data[p].j>>B->data[p].num;
	
while(pa<t1&&pb<t2){	
	switch(compare(A->data[pa].i,A->data[pa].j,B->data[pb].i,B->data[pb].j)){
		case 1:C->data[pc].i=B->data[pb].i;
		      C->data[pc].j=B->data[pb].j;
		      C->data[pc].num=B->data[pb].num;
		      pb++;
		      pc++;break;//A在B的后面，B加到C中 
		case -1:C->data[pc].i=A->data[pa].i;
		      C->data[pc].j=A->data[pa].j;
		      C->data[pc].num=A->data[pa].num;
		      pa++;
		      pc++;break;//B在A的后面 ，A加到C中 
		case 0:
			int res=A->data[pa].num+B->data[pb].num;
			if(res!=0) {
				C->data[pc].i=A->data[pa].i;
		      C->data[pc].j=A->data[pa].j;
		      C->data[pc].num=res;
		      pa++;
		      pb++;
		      pc++;
			}
			else{pa++;pb++;}
			break;
	}
}
     while(pa<t1) {C->data[pc].i=A->data[pa].i;
		      C->data[pc].j=A->data[pa].j;
		      C->data[pc].num=A->data[pa].num;
		      pa++;
		      pc++;}
     while(pb<t2) {
	 C->data[pc].i=B->data[pb].i;
		      C->data[pc].j=B->data[pb].j;
		      C->data[pc].num=B->data[pb].num;
		      pb++;
		      pc++;}
     C->tu=pc;
     
    for(int r=0;r<pc;r++)
    cout<<C->data[r].i<<" "<<C->data[r].j<<" "<<C->data[r].num<<endl;
return 0;}