压缩矩阵的快速转置

最新推荐文章于 2024-06-15 17:37:02 发布

青春期奥特曼不会爱上怪兽学姐

最新推荐文章于 2024-06-15 17:37:02 发布

阅读量142

点赞数 2

文章标签：矩阵算法 c++

本文链接：https://blog.csdn.net/2301_80467001/article/details/138869568

版权

#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;
}

通过对稀疏矩阵的压缩，减小了对空间的占用，并且使横变列，方便于矩阵的计算