数据结构 严蔚敏 稀疏矩阵的转置

#include <iostream>
#include <cstdio>
#define MAXN 1500
#define m 6
#define n 7
using namespace std;
typedef struct
{
    int i,j;
    int e;
}Triple;
typedef struct
{
    Triple data[MAXN+1];
    int mu,nu,tu;
}TSMatrix;
void Init( int s[m+1][n+1],TSMatrix &A)
{
    //int s[m][n];

    s[1][2]=12;
    s[1][3]=9;
    s[3][1]=-3;
    s[3][6]=14;
    s[4][3]=24;
    s[5][2]=18;
    s[6][1]=15;
    s[6][4]=-7;
    int k=1;
    for(int i=1;i<=m;i++)
        for(int j=1;j<=n;j++)
           if(s[i][j]!=0)
           {
               A.data[k].i=i;
               A.data[k].j=j;
               A.data[k].e=s[i][j];
               k++;
           }
     A.tu=k-1;
     A.mu=6;
     A.nu=7;
}
void Fast(TSMatrix M,TSMatrix &T)
{
    T.mu=M.nu;T.nu=M.mu;T.tu=M.tu;
    if(T.tu)
    {
        int col,q;
        int num[MAXN],cpot[MAXN];
        for( col=1;col<=M.nu;col++) num[col]=0;
        for(int t=1;t<=M.tu;t++) ++num[M.data[t].j];
        cpot[1]=1;
        for(col=2;col<=M.nu;col++) cpot[col]=cpot[col-1]+num[col-1];
        for(int p=1;p<=M.tu;p++)
        {
            col=M.data[p].j;
            q=cpot[col];
            T.data[q].i=M.data[p].i;
            T.data[q].j=M.data[p].j;
            T.data[q].e=M.data[p].e;
            cpot[col]++;
        }

    }
}
void print(TSMatrix &A)
{
    for(int s=1;s<=A.tu;s++)
            cout<<A.data[s].i<<" "<<A.data[s].j<<" "<<A.data[s].e<<endl;
}
void print2(TSMatrix &A)
{
    for(int s=1;s<=A.tu;s++)
            cout<<A.data[s].j<<" "<<A.data[s].i<<" "<<A.data[s].e<<endl;
}
int main()
{

    int s[m+1][n+1];
    TSMatrix M,T;
    Init(s,M);
    cout<<"M的稀疏矩阵:"<<endl;
    print(M);
    Fast(M,T);
    cout<<"T的稀疏矩阵:"<<endl;
    print2(T);
    return 0;
}```

猜你喜欢

转载自blog.csdn.net/curiosity9/article/details/83928482
今日推荐