数据结构作业之矩阵

矩阵乘法

本小结用数据结构简单实现了矩阵乘的基本算法。

 

头文件及结构体

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

#define ROWS 4
#define COLUMNS 5

/**
 * 二维数组
 */
typedef struct TwoDArray{
    int rows;
    int columns;
    int** elements;
}TwoDArray, *TwoDArrayPtr;

/**
 * 二维数组
 */
typedef struct TwoDStaticArray{
    int rows;
    int columns;
    int elements[ROWS][COLUMNS];
}TwoDStaticArray, *TwoDStaticArrayPtr;

初始化二维数组

/**
 * 初始化二维数组
 */
TwoDArrayPtr initTwoDArray(int paraRows,int paraColumns){
    int i;
    TwoDArrayPtr resultPtr=(TwoDArrayPtr)malloc(sizeof(struct TwoDArray));
    resultPtr->rows=paraRows;
    resultPtr->columns=paraColumns;
    resultPtr->elements=(int**)malloc(paraRows*sizeof(int*));
    for(i=0;i<paraRows;i++){
        resultPtr->elements[i]=(int*)malloc(paraColumns*sizeof(int));
    }

    return resultPtr;
}

/**
 * 随机初始化二维数组数据
 */
void randmizeTwoDArray(TwoDArrayPtr paraPtr, int paraLowerBound, int paraUpperBound){
    int i,j;
    for(i=0;i<paraPtr->rows;i++){
        for(j=0;j<paraPtr->columns;j++){
            paraPtr->elements[i][j]=rand()%(paraUpperBound-paraLowerBound)+paraLowerBound;
        }
    }
}

 打印二维数组

/**
 * 打印二维数组
 */
void printTwoDArray(TwoDArrayPtr paraPtr){
    int i,j;
    for(i=0;i<paraPtr->rows;i++){
        for(j=0;j<paraPtr->columns;j++){
           printf("%d, ", paraPtr->elements[i][j]);
        }

        printf("\r\n");
    }
}

矩阵乘算法

/**
 * 矩阵乘
 */
TwoDArrayPtr matrixMultiply(TwoDArrayPtr paraPtr1,TwoDArrayPtr paraPtr2){
    int i,j,k,sum;
    if(paraPtr1->columns!=paraPtr2->rows){
        printf("Matrices cannot be multiplied.\r\n");
		return NULL;
    }

    TwoDArrayPtr resultPtr=initTwoDArray(paraPtr1->rows,paraPtr2->columns);

    for (i = 0; i < paraPtr1->rows; i ++){
		for (j = 0; j < paraPtr2->columns; j ++) {
			sum = 0;
			for (k = 0; k < paraPtr1->columns; k ++) {
				sum += paraPtr1->elements[i][k] * paraPtr2->elements[k][j];
			}
			resultPtr->elements[i][j] = sum;
			printf("sum = %d, ", sum);
		}
	}

	return resultPtr;
}

测试函数及程序入口

/**
 * 测试二维数组的生成
 */
void twoDArrayTest(){
	TwoDArrayPtr tempPtr1, tempPtr2, tempPtr3;
	tempPtr1 = initTwoDArray(3, 2);
	randmizeTwoDArray(tempPtr1, 1, 5);
	printf("The first matrix:\r\n");
	printTwoDArray(tempPtr1);

	tempPtr2 = initTwoDArray(2, 4);
	randmizeTwoDArray(tempPtr2, 4, 9);
	printf("The second matrix:\r\n");
	printTwoDArray(tempPtr2);

	tempPtr3 = matrixMultiply(tempPtr1, tempPtr2);
	printf("The result:\r\n");
	printTwoDArray(tempPtr3);
}

/**
 * 初始化二维数组
 */
TwoDStaticArrayPtr initTwoDStaticArray(){
    int i,j;
    TwoDStaticArrayPtr resultPtr=(TwoDStaticArrayPtr)malloc(sizeof(struct TwoDStaticArray));
    resultPtr->rows=ROWS;
    resultPtr->columns=COLUMNS;
    for(i=0;i<ROWS;i++){
        for(j=0;j<COLUMNS;j++){
            resultPtr->elements[i][j]=i*10+j;
            // 打印地址
            printf("(%d, %d): %d; \r\n", i, j, &(resultPtr->elements[i][j]));
        }
    }

    return resultPtr;
}
/**
 * 程序入口
 */
int main(){
    twoDArrayTest();
    TwoDStaticArrayPtr tempPtr=initTwoDStaticArray();

    return 0;
}

 输出结果

The first matrix:
2, 4, 
3, 1,
2, 1,
The second matrix:
7, 7, 6, 8,
4, 4, 5, 6,
sum = 30, sum = 30, sum = 32, sum = 40, sum = 25, sum = 25, sum = 23, sum = 30, sum = 18, sum = 18, sum = 17, sum = 22, The result:
30, 30, 32, 40,
25, 25, 23, 30,
18, 18, 17, 22,
(0, 0): 2037192;
(0, 1): 2037196;
(0, 2): 2037200;
(0, 3): 2037204;
(0, 4): 2037208;
(1, 0): 2037212;
(1, 1): 2037216;
(1, 2): 2037220;
(1, 3): 2037224;
(1, 4): 2037228;
(2, 0): 2037232;
(2, 1): 2037236;
(2, 2): 2037240;
(2, 3): 2037244;
(2, 4): 2037248;
(3, 0): 2037252;
(3, 1): 2037256;
(3, 2): 2037260;
(3, 3): 2037264;
(3, 4): 2037268;

矩阵的压缩及转置

矩阵的压缩本质就是将矩阵的行与列数与数据一起存在一起，本节使用数组将3者存在一起。

在一些高阶矩阵中，非零元素非常少，此时如果使用二维数组将造成 储存空间的浪费，这时可只储存部分元素，从而提高储存空间的利用 率，通常的做法是为多个相同值的元素只分配一个储存单元，对值为 零的元素不分配储存单元。我们把非零元素个数远小于二维数组总元 素个数，或元素分布呈一定规律的（对称矩阵，三角矩阵，对角矩阵） 特殊矩阵压缩储存。

头文件及结构体

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

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){
	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];
	}

	return resultPtr;
}

打印压缩矩阵

/**
 * 打印压缩矩阵
 */
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);
	}
}

 转置压缩矩阵

/**
 * 转置压缩矩阵
 */
CompressedMatrixPtr transposeCompressedMatrix(CompressedMatrixPtr paraPtr){
    // Step 1. 申请空间
    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(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.计算偏移
	for(i = 0; i < paraPtr->numElements; i ++) {
		tempColumnCounts[paraPtr->elements[i].j] ++;
	}
	tempOffsets[0] = 0;
	for(i = 1; i <= paraPtr->columns; i ++){
		tempOffsets[i] = tempOffsets[i - 1] + tempColumnCounts[i - 1];
		printf("tempOffsets[%d] = %d \r\n", i, tempOffsets[i]);
	}

	//Step 3. 填充数据
	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]++;
	}

	return resultPtr;
}

 tempColumn用来计数不同列数下拥有的数据个数，tempOffsets用来记录压缩转置后的压缩矩阵的位置。

测试函数与程序入口

/**
 * 测试压缩矩阵功能
 */
void compressedMatrixTest(){
    CompressedMatrixPtr tempPtr1,tempPtr2;
    int i,j,tempElements;

    // 创建一个相同的矩阵
    tempElements = 4;
    int** tempMatrix1 = (int**)malloc(tempElements * sizeof(int*));
	for(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(i = 0; i < tempElements; i ++){
		for(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
tempOffsets[3] = 4
After transpose.
(0, 0): 2
(0, 2): 5
(1, 2): 6
(2, 0): 3