数据结构 稀疏矩阵相乘问题

#include

#include

#define OK 1

#define ERROR 0

#define MAXSIZE 25 //最多非0元素的个数

#define MAXR 5 //rpos所能处理的最大行数

#define MAXC 5 //系数矩阵相乘时，保留临时列结果的数组temp[MAXC] typedef struct NODE{ //定义稀疏矩阵结点

int i;

int j;

int data;

} Node;

typedef struct MATRIX{ //定义稀疏矩阵（可以快速访问）

int mu, nu, tu;

Node matrix[MAXSIZE+1];

int rpos[MAXR+1];

} Matrix;

int CreatSMatrix( Matrix* M ); //创建一个矩阵（由用户输入原始矩阵，转化为稀疏矩阵方式储存）

int Print( Matrix M ); //打印一个稀疏矩阵

int Mul_SMatrix( Matrix M, Matrix N, Matrix *Q); //两个稀疏矩阵相乘

main(){

printf("计科四班刘辉学号：41012169");

printf("\n");

printf("稀疏矩阵相乘");

printf("\n\n");

Matrix A1, A2, A3; //定义矩阵

CreatSMatrix( &A1 );

CreatSMatrix( &A2 );

if( A1.nu==A2.mu ){ //判断能否相乘

Mul_SMatrix( A1, A2, &A3 );

printf("两矩阵相乘得：\n"); Print(A3);

}

system("pause");

}

//稀疏矩阵相乘

int Mul_SMatrix( Matrix M, Matrix N, Matrix *Q)

{

int i,Mj;

int arow, Mlim, Nlim, Mcol, Nrow;

int ctemp[MAXC];

Q->tu=0;

//初始化Q

Q->mu=M.mu; Q->nu=M.nu;

if(M.tu*N.tu!=0){

//非零矩阵

for(arow=1; arow<=M.mu; arow++){

for(i=1; i<=M.nu; i++)//清空累加器

ctemp[i]=0;

Q->rpos[arow]=Q->tu+1; //给Q->rpos[]数组赋值

Mlim = arow

//M中第arow行在结点数组中的范围

for( Mcol=M.rpos[arow]; Mcol

//遍历M中第arow行的每一个j

Mj=M.matrix[Mcol].j;

Nlim = Mj

for( Nrow=N.rpos[Mj]; Nrow

ctemp[N.matrix[Nrow].j] += M.matrix[Mcol].data * N.matrix[Nrow].data;

}

for(i=1; i<=Q->nu; i++){//列号对应元素不为零，赋值

if( ctemp[i] ){

if( ++Q->tu > MAXSIZE )

return 0;

Q->matrix[Q->tu].i = arow;

Q->matrix[Q->tu].j = i;

Q->matrix[Q->tu].data = ctemp[i];

}

}

}

}

return 1;

}

//构建稀疏矩阵

int CreatSMatrix( Matrix* M ){

int temp, i,j;

printf("输入矩阵的行列数：");

scanf("%d,%d", &M->mu,&M->nu);

M->tu=0;

printf("按行序输入矩阵：\n");

for( i=1; i<=M->mu; i++){

M->rpos[i]=M->tu+1; //每计算完一行，给rpos[]赋值

for( j=1; j<=M->nu; j++){

scanf("%d",&temp );

if( temp ){ //非0值保存

M->matrix[M->tu+1].i= i;

M->matrix[M->tu+1].j= j;

M->matrix[M->tu+1].data=temp;

M->tu++;

}

}

}

return OK;

}

//打印稀疏矩阵

int Print( Matrix M){

int i;

if(M.tu==0){

printf("空矩阵\n\n");

return ERROR;

}

printf("i\tj\tdata\n");

for( i=1; i<=M.tu; i++ )

printf("%d\t%d\t%d\n", M.matrix[i].i,M.matrix[i].j,M.matrix[i].data);

return OK;

}

运行结果：