您的位置:首页 > 其它

稀疏矩阵的快速转置

2015-11-01 00:00 337 查看
#include<iostream>
#include<cstdio>
#define maxn 111
using namespace std;
int a[maxn][maxn],n,m,cnt=1;
typedef struct
{
int x,y,val;//x为横坐标,y为纵坐标,val为值
}matrix;
void Scan()//输入矩阵
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)  scanf("%d",&a[i][j]);
}
void Print() //打印转置前的矩阵
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
printf("%-3d",a[i][j]);
printf("\n");
}
}
void tranTothree(matrix A[]) //将原矩阵转化为三元组表
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
if(a[i][j]) A[cnt].x=i,A[cnt].y=j,A[cnt].val=a[i][j],cnt++; //如果元素不为0,则写入三元组表
A[0].x=n,A[0].y=m,A[0].val=cnt-1; //三元组第一结构体存储矩阵的行,列,三元组内容的个数
}
void Printthree(matrix A[])//打印三元组表
{
for(int i=1;i<=A[0].val;i++)
printf("x=%d,y=%d,val=%d\n",A[i].x,A[i].y,A[i].val);
}
void trans(matrix A[],matrix B[]) //矩阵转置
{
int x[maxn],y[maxn];//x数组记录原矩阵每一列的个数(不含零),y数组记录转置后的矩阵每一行的起始位置
fill(x,x+maxn,0);
fill(y,y+maxn,0);
y[0]=1;//从y[1]开始记录,y[0]空间不用,但也要加入总数
B[0].x=A[0].y,B[0].y=A[0].x,B[0].val=A[0].val; //将转置后的行列数赋给新的三元表
int nx=A[0].x,ny=A[0].y,ncnt=A[0].val,id;
for(int i=1;i<=ncnt;i++)  id=A[i].y,x[id]++; //找转置前每一列元素的个数,也就是转置后每一行的个数
for(int i=1;i<=m;i++)  y[i]=y[i-1]+x[i-1]; //扫一遍整个转置前整个列,记录转置后每一行的起始位置
for(int i=1;i<=ncnt;i++) //扫一遍原三元表,进行转置
{
id=A[i].y;
if(!y[id]) continue; //若个数为零,则不用进行存储
B[y[id]].x=A[i].y,B[y[id]].y=A[i].x,B[y[id]].val=A[i].val;
y[id]++;//起始位置加一
}
}
void Print_trans(matrix B[])//将转置后的三元表打印成矩阵
{
int cnt=1;
for(int i=1;i<=B[0].x;i++)
{
for(int j=1;j<=B[0].y;j++)
{
if(B[cnt].x==i&&B[cnt].y==j) printf("%-3d",B[cnt].val),cnt++;
else printf("%-3d",0);
}
printf("\n");
}
}
int main()
{
matrix A[maxn],B[maxn];
while(~scanf("%d%d",&n,&m))
{
Scan();
tranTothree(A);
//Printthree(A);
trans(A,B);
Printthree(B);
Print_trans(B);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: