稀疏矩阵相乘

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];
            }
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值