Description
假设稀疏矩阵A和B均以三元组表作为存储结构。试编写矩阵相加的程序,另设三元组表C存放结果矩阵。矩阵大小为m行n列(0<m,n<100)
Input
第一行输入t1,t2(0<t1,t2<100) ,t1和t2分别是矩阵A和B中非零元素的个数,后面t1+t2行分别输入A和B中的元素,用三元组表示。
Output
输出三元组表C。
-
Sample Input
3 3 1 2 3 3 2 1 3 4 2 1 1 4 3 2 5 3 4 1
-
Sample Output
1 1 4 1 2 3 3 2 6 3 4 3
#include<stdio.h>
#include<stdlib.h>
#define MAXSIZE 100
typedef struct {
int i,j; // 行下标,列下标
int e; // 非零元素值
}Triple;
typedef struct {
Triple data[MAXSIZE+1]; // 非零元三元组表,data[0]未用
int mu,nu,tu; // 矩阵的行数、列数和非零元个数
}TSMatrix;
TSMatrix AddTSM(TSMatrix A,TSMatrix B)
{
TSMatrix C;
int ai,bi,ci,aj,bj,cj,ap,bp,cp;
ap = bp = cp = 1;
C.mu = A.mu;
C.nu = A.nu;
while(ap <= A.tu && bp <= B.tu){
ai = A.data[ap].i;
bi = B.data[bp].i;
if(ai > bi){
ci = bi;
while(ci == B.data[bp].i){
C.data[cp].i = ci;
C.data[cp].j = B.data[bp].j;
C.data[cp].e = B.data[bp].e;
++bp;
++cp;
}
}
else if(ai < bi){
ci = ai;
while(ci == A.data[ap].i){
C.data[cp].i = ci;
C.data[cp].j = A.data[ap].j;
C.data[cp].e = A.data[ap].e;
++ap;
++cp;
}
}
else if(ai == bi){
ci = ai;
aj = A.data[ap].j;
bj = B.data[bp].j;
if(aj > bj){
C.data[cp].i = ci;
C.data[cp].j = bj;
C.data[cp].e = B.data[bp].e;
++cp;
++bp;
}
else if(aj < bj){
C.data[cp].i = ci;
C.data[cp].j = aj;
C.data[cp].e = A.data[ap].e;
++cp;
++ap;
}
else if(aj == bj){
if(A.data[ap].e + B.data[bp].e != 0){
C.data[cp].i = ci;
C.data[cp].j = aj;
C.data[cp].e = A.data[ap].e + B.data[bp].e;
++cp;
}
++ap;
++bp;
}
}
}
//以上为稀疏矩阵A或B中的元素完全加完的情况
//以下为稀疏矩阵A或B中的元素部分剩余的情况
while(ap <= A.tu){
C.data[cp].i = A.data[ap].i;
C.data[cp].j = A.data[ap].j;
C.data[cp].e = A.data[ap].e;
++cp;
++ap;
}
while(bp <= B.tu){
C.data[cp].i = B.data[bp].i;
C.data[cp].j = B.data[bp].j;
C.data[cp].e = B.data[bp].e;
++cp;
++bp;
}
C.tu = --cp;
return C;
}
int main()
{
TSMatrix A;
TSMatrix B;
int t1,t2,i;
scanf("%d %d",&t1,&t2);
A.tu=t1;
A.mu=100;
A.nu=100;
B.tu=t2;
B.mu=100;
B.nu=100;
for(i=1;i<=t1;i++)
{
scanf("%d %d %d",&A.data[i].i,&A.data[i].j,&A.data[i].e);
}
for(i=1;i<=t2;i++)
{
scanf("%d %d %d",&B.data[i].i,&B.data[i].j,&B.data[i].e);
}
TSMatrix C;
C=AddTSM(A,B);
for(i=1;i<=C.tu;i++)
{
printf("%d %d %d\n",C.data[i].i,C.data[i].j,C.data[i].e);
}
return 0;
}