マトリックス加算アルゴリズム
2618 ワード
疎行列AおよびBは、いずれも三元グループテーブルを記憶構造とするものとする.行列加算のアルゴリズムを書き出し,結果行列を三元グループテーブルCに格納する.
疎行列の三元グループ順序テーブルタイプTSMatrixの定義:
疎行列の三元グループ順序テーブルタイプ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;
}