#include<iostream>
using namespace std;
typedef int elem;
typedef struct Triple{
int i;
int j;
elem e;
}Triple,*TriplePtr;
typedef struct CompressedMatrix{
int rows,columns,numElements;
Triple* elements;
}CompressedMatrix,*CompressedMatrixPtr;
CompressedMatrixPtr initCompressedMatrix(int paraRows,int paraColumns,int paraElements,int** paraData){
CompressedMatrixPtr resultPtr=(CompressedMatrixPtr)malloc(sizeof(CompressedMatrix));
resultPtr->rows=paraRows;
resultPtr->columns=paraColumns;
resultPtr->numElements=paraElements;
resultPtr->elements=(TriplePtr)malloc(sizeof(Triple)*paraElements);
for(int i=0;i<paraElements;i++){
resultPtr->elements[i].i=paraData[i][0];
resultPtr->elements[i].j=paraData[i][1];
resultPtr->elements[i].e=paraData[i][2];
}
return resultPtr;
}
void printCompressedMatrix(CompressedMatrixPtr paraPtr){
for(int i=0;i<paraPtr->numElements;i++){
printf("(%d, %d): %d\r\n",paraPtr->elements[i].i,paraPtr->elements[i].j,paraPtr->elements[i].e);
}
}
CompressedMatrixPtr transposeCompressedMatrix(CompressedMatrixPtr paraPtr){
int i,tempColumn,tempPosition;
int *tempColumnCounts=(int*)malloc(paraPtr->columns*sizeof(int));
int *tempOffsets=(int*)malloc(paraPtr->columns*sizeof(int));
for(i=0;i<paraPtr->columns;i++){
tempColumnCounts[i]=0;
}
CompressedMatrixPtr resultPtr=(CompressedMatrixPtr)malloc(sizeof(CompressedMatrix));
resultPtr->rows=paraPtr->columns;
resultPtr->columns=paraPtr->rows;
resultPtr->numElements=paraPtr->numElements;
resultPtr->elements=(TriplePtr)malloc(paraPtr->numElements*sizeof(Triple));
for(int i=0;i<paraPtr->numElements;i++) {
tempColumnCounts[paraPtr->elements[i].j]++;
}
tempOffsets[0]=0;
for(int i=1;i<paraPtr->columns;i++){
tempOffsets[i]=tempOffsets[i-1]+tempColumnCounts[i-1];
printf("tempOffsets[%d] = %d \r\n",i,tempOffsets[i]);
}
for(int i=0;i<paraPtr->numElements;i++){
tempColumn=paraPtr->elements[i].j;
tempPosition=tempOffsets[tempColumn];
resultPtr->elements[tempPosition].i=paraPtr->elements[i].j;
resultPtr->elements[tempPosition].j=paraPtr->elements[i].i;
resultPtr->elements[tempPosition].e=paraPtr->elements[i].e;
tempOffsets[tempColumn]++;
}
return resultPtr;
}
void compressedMatrixTest(){
CompressedMatrixPtr tempPtr1, tempPtr2;
int tempElements=4;
int** tempMatrix1=(int**)malloc(tempElements*sizeof(int*));
for(int i=0;i<tempElements;i++){
tempMatrix1[i]=(int*)malloc(3*sizeof(int));
}
int tempMatrix2[4][3]={{0, 0, 2}, {0, 2, 3}, {2, 0, 5}, {2, 1, 6}};
for(int i=0;i<tempElements;i++){
for(int j=0;j<3;j++){
tempMatrix1[i][j]=tempMatrix2[i][j];
}
}
tempPtr1=initCompressedMatrix(2,3,4,tempMatrix1);
printf("After initialization.\r\n");
printCompressedMatrix(tempPtr1);
tempPtr2=transposeCompressedMatrix(tempPtr1);
printf("After transpose.\r\n");
printCompressedMatrix(tempPtr2);
}
int main(){
compressedMatrixTest();
return 0;
}
通过阅读和理解上述代码,我学到了关于稀疏矩阵的压缩存储和转置操作的一些知识。
首先,代码中定义了两个结构体:Triple和CompressedMatrix,分别用来表示矩阵中的非零元素和压缩存储的稀疏矩阵。Triple结构体包含了非零元素的行、列和值,而CompressedMatrix结构体则包含了矩阵的行数、列数、非零元素个数以及一个指向Triple数组的指针。
在initCompressedMatrix函数中,实现了初始化稀疏矩阵的功能。函数通过传入的参数构建了一个CompressedMatrix对象,并将非零元素的信息存储在Triple数组中。
接着,在transposeCompressedMatrix函数中,实现了稀疏矩阵的转置操作。函数首先创建了一个新的CompressedMatrix对象用于存储转置后的矩阵,然后通过临时数组tempColumnCounts和tempOffsets来辅助计算转置后每列非零元素的位置。最后,将原矩阵的非零元素按列重新排列到新矩阵中,实现了转置操作。
最后,在compressedMatrixTest函数中,进行了稀疏矩阵的初始化和转置操作的测试。通过打印输出,可以看到稀疏矩阵在初始化和转置后的结果,验证了代码的正确性。
总的来说,通过阅读这段代码,我学习到了如何使用结构体和指针来实现稀疏矩阵的压缩存储和转置操作,加深了对C++中指针和动态内存分配的理解,同时也对稀疏矩阵的基本概念有了更深入的认识。这段代码为我提供了一个很好的学习机会,让我更加熟悉和理解了这些概念和操作。