稀疏矩阵加法,实现C=A+B

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;
}

猜你喜欢

转载自blog.csdn.net/zhao2018/article/details/80112873