5.2 压缩矩阵的转置

#include <stdio.h>
#include <malloc.h>

typedef int elem;

/**
 *A triple for row index,column index,and data.
 */
typedef struct Triple{
	int i;
	int j;
	elem e;
}Triple,*TriplePtr;

/**
 *A triple for row index,column index,and data.
 */
typedef struct CompressedMatrix{
	int rows,columns,numElements;
	Triple* elements;
}CompressedMatrix,*CompressedMatrixPtr;

/**
 *Initialize a compressed matrix.
 */
CompressedMatrixPtr initCompressedMatrix(int paraRows,int paraColumns,int paraElements,int** paraData){
	int i;
	CompressedMatrixPtr resultPtr = (CompressedMatrixPtr)malloc(sizeof(struct CompressedMatrix));
	resultPtr->rows = paraRows;
	resultPtr->columns  =paraColumns;
	resultPtr->numElements = paraElements;
	resultPtr->elements = (TriplePtr)malloc(paraElements * sizeof(struct Triple));
	
	for(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];
       }//Of for i.

return resultPtr;
}//Of initCompressedMatrix.

/**
 *Print the compressed matrix.
 */
void printCompressedMatrix(CompressedMatrixPtr paraPtr){
	int i;
	for(i = 0;i < paraPtr->numElements;i ++){
		printf("(%d,%d):%d\r\n",paraPtr->elements[i].i,paraPtr->elements[i].j,paraPtr->elements[i].e);
	}//Of for i
}//Of printCompressedMatrix.

/**
 *Transpose a compressed matrix.
 */
CompressedMatrixPtr transposeCompressedMatrix(CompressedMatrixPtr paraPtr){
	//Step 1. Allocate space.
	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;
	}//Of for i.
	
	CompressedMatrixPtr resultPtr = (CompressedMatrixPtr)malloc(sizeof(struct CompressedMatrix));
	resultPtr->rows = paraPtr->columns;
	resultPtr->columns = paraPtr->rows;
	resultPtr->numElements = paraPtr->numElements;
	
	resultPtr->elements = (TriplePtr)malloc(paraPtr->numElements * sizeof(struct Triple));
	
	//Step 2. One scan to calculate offsets.
	for(i = 0;i < paraPtr->numElements;i ++){
		tempColumnCounts[paraPtr->elements[i].j] ++;
	} //Of for i.
	tempOffsets[0] = 0;
	for(i = 0;i < paraPtr->columns;i ++){
		tempOffsets[i] = tempOffsets[i - 1] + tempColumnCounts[i - 1];
		printf("tempOffsets[%d] = %d \r\n",i,tempOffsets[i]);
	}//Of for i.
	
	//step 3. Another scan to fill data.
	for(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]++;
	}//Of for i.
	
	return resultPtr;
}//Of tempsposeCompressedMatrix.

/**
 *Test the compressed matrix.
 */
void compressedMatrixTest(){
	CompressedMatrixPtr tempPtr1,tempPtr2;
	int i,j,tempElements;
	
	//Construct the first sample matrix.
	tempElements = 4;
	int** tempMatrix1 =  (int**)malloc(tempElements * sizeof(int*));
	for(i = 0;i < tempElements;i ++){
		tempMatrix1[i] = (int*)malloc(3 * sizeof(int));
	} //Of for i.
	
	int tempMatrix2[4][3] = {{0,0,2},{0,2,3},{2,0,5},{2,1,6}};
	for(i = 0;i < tempElements;i ++){
		for(j = 0;j < 3;j ++){
			tempMatrix1[i][j] = tempMatrix2[i][j];
		}//Of for j.
	}//Of for i.
	
	tempPtr1 = initCompressedMatrix(2,3,4,tempMatrix1);
	
	printf("After initialization.\r\n");
	printCompressedMatrix(tempPtr1);
	
	tempPtr2 = transposeCompressedMatrix(tempPtr1);
	printf("After transpose.\r\n");
	printCompressedMatrix(tempPtr2);
}//Of main.

/**
 *The entrance.
 */
int main(){
	compressedMatrixTest();
	
	return 1;
}//Of main.

结果：