矩阵相加的算法(存储结构为三元组表)

假设稀疏矩阵A和B均以三元组表作为存储结构。试写出矩阵相加的算法,另设三元组表C存放结果矩阵。

稀疏矩阵的三元组顺序表类型TSMatrix的定义:

#define MAXSIZE 20 // 非零元个数的最大值
typedef struct {
  int    i,j; // 行下标,列下标
  ElemType e; // 非零元素值
}Triple;
 
typedef struct  {
  Triple data[MAXSIZE+1]; // 非零元三元组表,data[0]未用
  int mu,nu,tu; // 矩阵的行数、列数和非零元个数
}TSMatrix;
实现函数如下:

Status AddTSM(TSMatrix A,TSMatrix B,TSMatrix &C)
/* 三元组表示的稀疏矩阵加法: C=A+B */
{
    int ai,bi,ci,aj,bj,cj,ap,bp,cp;
    ap = bp = cp = 1;
    if(A.mu != B.mu || A.nu != B.nu){
        return ERROR;
    }
    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 OK;
}


  • 27
    点赞
  • 97
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值