数据结构实验之数组三:快速转置
Time Limit: 1000ms Memory limit: 65536K 有疑问?点这里^_^
题目描述
转置运算是一种最简单的矩阵运算,对于一个m*n的矩阵M( 1 = < m < = 10000,1 = < n < = 10000 ),它的转置矩阵T是一个n*m的矩阵,且T( i , j )=M( j , i )。显然,一个稀疏矩阵的转置仍然是稀疏矩阵。你的任务是对给定一个m*n的稀疏矩阵( m , n < = 10000 ),求该矩阵的转置矩阵并输出。矩阵M和转置后的矩阵T如下图示例所示。
稀疏矩阵M 稀疏矩阵T
输入
连续输入多组数据,每组数据的第一行是三个整数mu, nu, tu(tu <= 50),分别表示稀疏矩阵的行数、列数和矩阵中非零元素的个数,随后tu行输入稀疏矩阵的非零元素所在的行、列值和非零元素的值,同一行数据之间用空格间隔。(矩阵以行序为主序)
输出
输出转置后的稀疏矩阵的三元组顺序表表示。
示例输入
3 5 5
1 2 14
1 5 -5
2 2 -7
3 1 36
3 4 28
示例输出
1 3 36
2 1 14
2 2 -7
4 3 28
5 1 -5
提示
来源
xam
#include<bits/stdc++.h>
using namespace std;
#define MAXSIZE 10010
typedef struct
{
int i,j;
int e;
}Triple;
typedef struct
{
Triple data[MAXSIZE];
int mu,nu,tu;
}TSMatrix;
int num[MAXSIZE],cpot[MAXSIZE];
void FastTransposeSMatrix(TSMatrix M,TSMatrix T)//快速转置函数
{
T.mu=M.mu; T.nu=M.nu; T.tu=M.tu;
int col,t,p,q;
if(T.tu)
{
for(col=1;col<=M.nu;++col) num[col]=0;
for(t=1;t<=M.tu;++t) ++num[M.data[t].j];//求M中每一列含非零元素的个数
cpot[1]=1;
//求第col列中第一个非零元素在T.data中的序号
for(col=2;col<=M.nu;++col) cpot[col]=cpot[col-1]+num[col-1];
for(p=1;p<=M.tu;++p)
{
col=M.data[p].j;
q=cpot[col];
T.data[q].i=M.data[p].j;
T.data[q].j=M.data[p].i;
T.data[q].e=M.data[p].e;
++cpot[col];
}
}
//输出T.data
for(int j=1;j<=T.tu;j++)
{
cout<<T.data[j].i<<" "<<T.data[j].j<<" "<<T.data[j].e<<endl;
}
}
int main()
{
TSMatrix M,T;
while(cin>>M.mu>>M.nu>>M.tu)
{
for(int i=1;i<=M.tu;i++)
{
cin>>M.data[i].i>>M.data[i].j>>M.data[i].e;
}
FastTransposeSMatrix(M,T);
}
return 0;
}