数据结构作业12--压缩矩阵与快速转置

代码本体

#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

代码说明

1. 读数据有点麻烦, 如果原先存在于文件中应该容易些.
2. 转置时间复杂度低.
3. 本来想再写一个乘法的, 算了, 有能力的同学自己去摸索. 但要注意, 两个稀疏矩阵相乘之后, 一般不是稀疏矩阵, 所以结果也不用压缩矩阵来存储. 同时, 应该在结构体中多定义一个 offsets 数组, 方便定位和计算.
4. 这个代码用于学习对数组下标的操控, 也有利于进一步体会内存的组织.

图示