Description
输入两个稀疏矩阵,输出它们相加的结果。
Input
第一行输入四个正整数,分别是两个矩阵的行m、列n、第一个矩阵的非零元素的个数t1和第二个矩阵的非零元素的个数t2。
接下来的t1+t2行是三元组,分别是第一个矩阵的数据和第二个矩阵的数据。三元组的第一个元素表示行号,第二个元素表示列号,第三个元素是该项的值。
Output
输出相加后的矩阵三元组。
-
Sample Input
3 4 3 2 1 1 1 1 3 1 2 2 2 1 2 1 2 2 3
-
Sample Output
1 1 1 1 2 1 1 3 1 2 2 5
#include<stdio.h> #include<stdlib.h> #define MAXSIZE 100 typedef struct { int i,j; int e; }Triple; typedef struct { Triple data[MAXSIZE+1]; 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 n,m,t1,t2,i; scanf("%d %d %d %d",&m,&n,&t1,&t2); A.tu=t1; A.mu=m; A.nu=n; B.tu=t2; B.mu=m; B.nu=n; 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; }