1483 稀疏矩阵的压缩存储与相乘

//稀疏矩阵乘法 行逻辑链接的三元组顺序表
// 3 4
// 6
// 1 1 3
// 1 2 2
// 1 3 3
// 1 4 5
// 2 2 -1
// 3 1 2
// 4 2
// 5
// 1 2 2
// 2 1 1
// 3 1 -2
// 3 2 4
// 4 1 2
#include <iostream>
#include <cstring>
#define  MAX_SIZE 2500+1
#define  MAX_RC   2500+1
using namespace std;
typedef int elemType;
typedef struct {
    int i,j;
    elemType e;
}Triple;
typedef struct{
    Triple data[MAX_SIZE];
    int    rowPos[MAX_RC];//各行第一个元素的秩表
    int    mu,nu,tu;
}TripleSpareMatrix;
void creatTripleSpareMatrix(TripleSpareMatrix &tSM)
{
    scanf("%d%d%d",&tSM.mu,&tSM.nu,&tSM.tu);
    int rowNum[tSM.mu+1];
    for (int i = 0; i < tSM.mu+1; ++i) rowNum[i]=0;//行元素数目累加器初始化
    for(int k=1;k<=tSM.tu;k++)
    {
        scanf("%d%d%d",&tSM.data[k].i,&tSM.data[k].j,&tSM.data[k].e);
        rowNum[tSM.data[k].i]++;
    }
    tSM.rowPos[1]=1;
    for (int p = 2; p <=tSM.mu+1; ++p)//对不存在的第[tSm.mu+1]行统一处理
        tSM.rowPos[p]=tSM.rowPos[p-1]+rowNum[p-1];
}
void multMatrix(TripleSpareMatrix tSM1,TripleSpareMatrix tSM2,TripleSpareMatrix &tSM3)
{
    tSM3.mu=tSM1.mu; tSM3.nu=tSM2.nu; tSM3.tu=0;//tSM3初始化
    for (int row = 1; row <= tSM1.mu; ++row)//遍历tSM3（同tSM1）的每一行
    {
        int colNum[tSM2.nu+1];
        for (int i = 0; i < tSM2.nu+1; ++i) colNum[i]=0;//当前行个元素累加器初始化
        tSM3.rowPos[row]=tSM3.tu+1;//
        for (int p = tSM1.rowPos[row]; p <tSM1.rowPos[row+1]; ++p)//遍历当前的row行的每一个非零元
            for (int q=tSM2.rowPos[tSM1.data[p].j]; q<tSM2.rowPos[tSM1.data[p].j+1]; ++q)//遍历与tSM1当前行的对应tSM2行（tSM1.data[p].j==tSM2.data[q].i）
                colNum[tSM2.data[q].j]+=tSM1.data[p].e*tSM2.data[q].e;//累加
        for (int col = 1; col <= tSM2.nu; ++col)//压缩存储非零元
            if(colNum[col])
            {
                tSM3.tu++;
                tSM3.data[tSM3.tu].i=row; tSM3.data[tSM3.tu].j=col; tSM3.data[tSM3.tu].e=colNum[col];
            }
    }
}
void printMatrix(TripleSpareMatrix tSM)
{
    for (int i = 1; i <=tSM.tu; ++i)
        printf("%d %d %d
",tSM.data[i].i,tSM.data[i].j,tSM.data[i].e );
}
int main(int argc, char const *argv[])
{
    #ifndef _OJ_  //ONLINE_JUDGE
    freopen("input.txt", "r", stdin);
    #endif
    TripleSpareMatrix tSM1,tSM2,tSM3;
    creatTripleSpareMatrix(tSM1);
    creatTripleSpareMatrix(tSM2);
    multMatrix(tSM1,tSM2,tSM3);
    printMatrix(tSM3);
    return 0;
}