SCAU_OJ 18722 稀疏矩阵的运算

18722 稀疏矩阵的运算

时间限制:1000MS 代码长度限制:10KB
提交次数:0 通过次数:0

题型: 编程题 语言: G++;GCC

Description

稀疏矩阵的压缩存储原则：只存矩阵的行列数和每个非零元的行列下标及其值。
例如下图的矩阵M由行列数（6,7）和三元组表{(1,2,12), (1,3,9), (3,1,-3),(3,6,14),(4,3,24),(5,2,18), (6,1,15), (6,4,-7) }唯一确定。
问题描述：已知一个稀疏矩阵的三元组表，使用快速转置算法求其转置矩阵的三元组表，三元组表要按行优先的方式存储。

输入格式

第一行三个整数n,m,k。n,m代表矩阵A的行列数（0<=n,m<=1000000）,k为三元组表中元素的个数。
此后为k行，每行3个整数a，b，c，分别代表元素的行号，列号和值。数据确保按行优先给出。（0<=k<=10000）

输出格式

输出为k行，即转置矩阵的三元组表，三元组表要按行优先显示。

输入样例

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

输出样例

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

题解

这题明显不适用于先普通转置再用排序算法甚至sort()。
根据空间换时间的思想，可以想到如果只扫描一遍三元组就可将转置后的三元组完全填满，那这一步就是O(n)。
因此我们可以借助一个辅助数组pos[]，来记录转置后的每一行的首个非零元素在三元组中的位置。（pos的下标为转置后的矩阵对应的行号，即转置前的列号）
不难想到，pos[i]=pos[i-1]+转置后上一行的元素数量
故再搞一个num[]，来记录转置后每一行的元素数量
则pos[i]=pos[i-1]+num[i-1]

#include <iostream>

using namespace std;

int num[1000005];
int pos[1000005];
struct M
{
    int row;
    int col;
    int data;
}matrix[10005],trans[10005];

int n,m,k;
int main()
{
    cin>>n>>m>>k;
    int a,b,c;
    for(int i=1;i<=k;i++)
    {
        cin>>a>>b>>c;
        matrix[i].row=a;
        matrix[i].col=b;
        matrix[i].data=c;
    }
    for(int i=1;i<=k;i++)
    {
        int p=matrix[i].col;
        num[p]+=1;
    }
    pos[0]=1;
    for(int i=1;i<=m;i++)
    {
        pos[i]=pos[i-1]+num[i-1];
    }
    for(int i=1;i<=k;i++)
    {
        int j=matrix[i].col;
        int p=pos[j];
        pos[j]++;
        trans[p].row=matrix[i].col;
        trans[p].col=matrix[i].row;
        trans[p].data=matrix[i].data;
    }
    for(int i=1;i<=k;i++)
    {
        cout<<trans[i].row<<" "<<trans[i].col<<" "<<trans[i].data<<endl;
    }

    return 0;
}