代码本体
#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;
}
运行结果
After initialization.
(0, 0): 2
(0, 2): 3
(2, 0): 5
(2, 1): 6
tempOffsets[1] = 2
tempOffsets[2] = 3
After transpose.
(0, 0): 2
(0, 2): 5
(1, 2): 6
(2, 0): 3
Press any key to continue
-
代码说明
- 读数据有点麻烦, 如果原先存在于文件中应该容易些.
- 转置时间复杂度低.
- 本来想再写一个乘法的, 算了, 有能力的同学自己去摸索. 但要注意, 两个稀疏矩阵相乘之后, 一般不是稀疏矩阵, 所以结果也不用压缩矩阵来存储. 同时, 应该在结构体中多定义一个 offsets 数组, 方便定位和计算.
- 这个代码用于学习对数组下标的操控, 也有利于进一步体会内存的组织.
图示