矩阵的快速转置算法

题目信息

数据压缩是提高传输、存储效率一种技术。本实验要求实现三元组顺序表表示下的矩阵快速转置算法。

输入

稀疏矩阵的行数、列数、非零元个数(三个数都大于0)
以行为主序输入稀疏矩阵三元组表

输出

辅助数组num[ ]
辅助数组cpot[ ]
以行为主序输出对应的转置矩阵三元组表

测试样例

6 7 8
1 2 12
1 3 9
3 1 -3
3 6 14
4 3 24
5 2 18
6 1 15
6 4 -7

num:2,2,2,1,0,1,0,
cpot:1,3,5,7,8,8,9,
1,3,-3
1,6,15
2,1,12
2,5,18
3,1,9
3,4,24
4,6,-7
6,3,14

解答

#include <iostream>

using namespace std;

struct Node
{
    int x;
    int y;
    int data;
};

int main()
{
    //freopen("/Users/zhj/Downloads/test.txt", "r", stdin);
    int N, M, NUM;
    cin >> N >> M >> NUM;

    Node Matrix[NUM];//原矩阵
    for (int i = 0; i < NUM; i++)
    {
        cin >> Matrix[i].x >> Matrix[i].y >> Matrix[i].data;
    }

    int num[M];
    for (int i = 0; i < M; i++)
    {
        num[i] = 0;
    }
    for (int i = 0; i < NUM; i++)
    {
        num[Matrix[i].y - 1]++;
    }
    cout << "num:";
    for (int i = 0; i < M; i++)
    {
        cout << num[i] << ",";
    }
    cout << endl;

    int cpot[M];
    cpot[0] = 0;
    for (int i = 1; i < M; i++)
    {
        cpot[i] = cpot[i - 1] + num[i - 1];
    }
    cout << "cpot:";
    for (int i = 0; i < M; i++)
    {
        cout << cpot[i] + 1 << ",";
    }
    cout << endl;

    Node TransMatrix[NUM];//转置后矩阵
    int count = 0, flag;
    for (int i = 0; i < NUM; i++)
    {//以原矩阵作为依据
        flag = Matrix[i].y-1;//以列作为标志
        TransMatrix[cpot[flag]].x = Matrix[i].y;
        TransMatrix[cpot[flag]].y = Matrix[i].x;
        TransMatrix[cpot[flag]].data = Matrix[i].data;
        cpot[flag]++;//在列中相应的位置需要后移
    }

    for (int i = 0; i < NUM; i++)
    {
        cout << TransMatrix[i].x << "," << TransMatrix[i].y << "," << TransMatrix[i].data << endl;
    }
    return 0;
}

补充

在此附设num和cpot两个向量.
num[col]表示矩阵M中第col列中非零元的个数
cpot[col]指示M中第col列的第一个非零元在b.data中的恰当位置