稀疏矩阵的快速转置

来源:互联网 发布:软件开发语言 编辑:程序博客网 时间:2024/04/28 06:45
#include<iostream>#include<cstdio>#define maxn 111using 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;}

0 0
原创粉丝点击