矩阵快速转置

/*
矩阵的快速转置
*/

#include<iostream>
using namespace std;

#define MAX 100

typedef struct mode{
    int x,y;
    int value;
}SRC;
typedef struct node_1{
    SRC data[MAX+1];
    int mu,nu,tu;  /*分别表示矩阵的行数 列数和非零元素的数目*/
}SRCD;

void CreateSRCD(int arr[][5] ,int m,int n,SRCD* &srcd)
{
    srcd->tu = 0;
    int k=0;
    for(int i=1;i<=m;i++){
        for(int j=1;j<=n;j++){
            if(arr[i][j] != 0){
                k++; 
                srcd->data[k].x = i;
                srcd->data[k].y = j;
                srcd->data[k].value = arr[i][j];
                srcd->tu++;
            }
        }
    }
    srcd->mu = m;
    srcd->nu = n;
}

void QuickSz(SRCD* src,SRCD* &dest)
{
    if(src == NULL || src->tu == 0)  return ;
    int *num = new int[src->nu+1];  /*存放原始矩阵的每一列的第一个非零元素的数目*/
    for(int i=1;i<=src->nu;i++)   num[i] = 0;  /*初始化原始的矩阵中的每一列的第一个非零元素数目*/
    for(int i=1;i<=src->tu;i++)    ++num[src->data[i].y];   /*计算src中国每一列的非零元素的数目*/
    int *pos = new int[src->tu+1];  /*存放src中的每一列的第一个非零元素的位置*/
    pos[1]=1;  /*第一列第一个非零元素将会在转置后的新的三元组的第一个位置出现*/
    for(int i=2;i<=src->nu;i++)  pos[i] = pos[i-1]+num[i-1];  /*处第列外的每一列的第一个非零元素的位置是上一列中的第一个非零元素的位置加上上一列的非零元素的个数*/
    for(int i=1;i<=src->tu;i++){
        int k = src->data[i].y;
        int l = pos[k];
        dest->data[l].x = src->data[i].y;
        dest->data[l].y = src->data[i].x;
        dest->data[l].value = src->data[i].value;
        ++pos[k];
    }
    dest->mu = src->nu;
    dest->nu = src->mu;
    dest->tu = src->tu;
}

void display(SRCD* src)
{
    for(int i=1;i<=src->tu;i++){
        cout<<src->data[i].x<<" "<<src->data[i].y <<" "<< src->data[i].value<<endl;
    }
}

int main()
{
    int arr[4][5];
    for(int i=1;i<=3;i++){
        for(int j=1;j<=4;j++){
            cin>>arr[i][j];
        }
    }
    SRCD *src = new SRCD;
    SRCD *dest = new SRCD;
    CreateSRCD(arr,3,4,src);
    display(src);
    cout<<"------------------------我是分割线-------------------------------------"<<endl;
    QuickSz(src,dest);
    display(dest);
    system("pause");
    return 0;
}