bool MultSMatrix(RLSMatrix M, RLSMatrix N, RLSMatrix &Q)
{
if (M.data[0].j != N.data[0].i)//data[0]存储了矩阵的行数、列数、非零元个数,这里是M的列数不等于N的行数,根据矩阵相乘的规则,不能相乘
return false;
Q.data[0].i = M.data[0].i;
Q.data[0].j = N.data[0].j;
Q.data[0].value = 0;
if (M.data[0].value * N.data[0].value == 0)//和零矩阵相乘得到零矩阵
return true;
for (int row = 1; row <= M.data[0].i; row++)//逐行处理
{
int temp[N.data[0].j] = {};//当前行元素累加器清零
Q.rpos[row] = Q.data[0].value + 1;//当前行第一个非零元的位置
int last1;//记录M下一行第一个元素的位置,用来做边界的
if (row < M.data[0].i)
last1 = M.rpos[row+1];
else
last1 = M.data[0].value + 1;
for (int k = M.rpos[row]; k < last1; k++)
{
int q = M.data[k].j;//N中的行号
int last2;//记录N下一行第一个元素的位置,用来做边界的
if (q < N.data[0].i)
last2 = N.rpos[q+1];
else
last2 = N.data[0].value + 1;
for (int j = N.rpos[q]; j < last2; t++)
{
int col = N.data[j].j;//乘积元素在Q中的列号
temp[col] += M.data[k].value * N.data[j].value;
}
}//求得Q中第row行的非零元
for (int col = 1; col <= Q.data[0].j; col++)
{
if(temp[col] != 0)
{
if (++Q.data[0].value >= MAX)
return false;
int i = Q.data[0].value;
Q.data[i].i = row;
Q.data[i].j = col;
Q.data[i].value = temp[col];
}
}
}
}
稀疏矩阵相乘
最新推荐文章于 2022-04-21 00:29:43 发布