oj稀疏矩阵的运算 (0n算法)
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可惜复杂度有nlogn。
反思:
在老师的教导下,引进新方法,
- 通过 num【】 记录列号,
- 之后再利用前缀和的思想把每一列的元素映射到 cpot【】 里。
- 最后再利用b【cpot【a【i】.y】++】=a[i].
AC
#include <iostream>
#define For(i,x,y) for(register int i=(x); i<=(y); i++)
using namespace std;
const int maxn=1e6+10;
struct point
{
int x,y,v;
}a[maxn],b[maxn];
int n,m,k;
int num[maxn],cpot[maxn];
int main()
{
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
cin>>n>>m>>k;
For(i,1,k)cin>>a[i].x>>a[i].y>>a[i].v,num[a[i].y]++;
cpot[1]=1;
For(i,2,m)cpot[i]=cpot[i-1]+num[i-1];//
For(i,1,k)b[cpot[a[i].y]++]=a[i];//这里应该是直接读取了编号
For(i,1,k)cout<<b[i].y<<' '<<b[i].x<<' '<<b[i].v<<endl;//
return 0;
}