【数据结构】数组和字符串(一):矩阵的数组表示
矩阵是以按行优先次序将所有矩阵元素存放在一个一维数组中。但是对于特殊矩阵,如对称矩阵、三角矩阵、对角矩阵和稀疏矩阵等, 如果用这种方式存储,会出现大量存储空间存放重复信息或零元素的情况,这样会造成很大的空间浪费。为节约存储空间和算法(程序)运行时间,通常会采用压缩存储的方法。
对于一个n×n维的对角矩阵M,由于非主对角线上的元素都为零,只需存储其n个对角元素的值即可。可以使用一维数组d[n]来压缩存储对角矩阵,其中d[i-1] (1≤ i ≤ n)存储M(i, i)的值。这种压缩存储方式可以显著减少存储空间的使用量,尤其在矩阵规模较大时效果更为明显。同时,在对角矩阵的运算中,由于非主对角线上的元素都为零,可以通过直接访问压缩后的数据来提高算法的效率。
typedef struct {
int size; // 矩阵的维度
int diagonal[MAX_SIZE]; // 存储对角元素的数组
} DiagonalMatrix;
DiagonalMatrix
结构体定义了对角矩阵的结构,包括矩阵的维度 size
和存储对角元素的数组 diagonal
。
void initialize(DiagonalMatrix *matrix, int size) {
matrix->size = size;
// 初始化对角元素数组
for (int i = 0; i < size; i++) {
matrix->diagonal[i] = 0;
}
}
initialize
函数用于初始化对角矩阵,接受一个指向 DiagonalMatrix
结构体的指针和矩阵的维度作为参数,在函数内部将矩阵的维度存储到 size
成员变量中,并将对角元素数组的所有元素初始化为0。
void setElement(DiagonalMatrix *matrix, int row, int col, int value) {
if (row != col) {
printf("Error: Only diagonal elements can be set.\n");
} else if (row < 0 || row >= matrix->size || col < 0 || col >= matrix->size) {
printf("Error: Invalid row or column index.\n");
} else {
matrix->diagonal[row] = value;
}
}
setElement
函数用于设置对角矩阵中指定位置的元素值,接受一个指向 DiagonalMatrix
结构体的指针、行索引、列索引和要设置的值作为参数。
int getElement(DiagonalMatrix *matrix, int row, int col) {
if (row < 0 || row >= matrix->size || col < 0 || col >= matrix->size) {
printf("Error: Invalid row or column index.\n");
return 0;
} else if (row != col) {
return 0;
} else {
return matrix->diagonal[row];
}
}
getElement
函数用于获取对角矩阵中指定位置的元素值,接受一个指向 DiagonalMatrix
结构体的指针、行索引和列索引作为参数。
void printMatrix(DiagonalMatrix *matrix) {
for (int i = 0; i < matrix->size; i++) {
for (int j = 0; j < matrix->size; j++) {
printf("%d ", getElement(matrix,i,j));
}
printf("\n");
}
}
printMatrix
函数用于打印对角矩阵,接受一个指向 DiagonalMatrix
结构体的指针作为参数。函数通过两层循环遍历矩阵的每个位置,调用getElement
函数获取并打印元素的值。
int main() {
DiagonalMatrix matrix;
int size = 6;
initialize(&matrix, size);
// 读入数据并设置对角矩阵的元素
for (int i = 0; i < size; i++) {
int value;
printf("\nEnter the value for element [%d][%d]: ", i, i);
scanf("%d", &value);
setElement(&matrix, i, i, value);
}
printf("Diagonal Matrix:\n");
printMatrix(&matrix);
return 0;
}
在 main
函数中,首先定义了一个 DiagonalMatrix
类型的变量 matrix
和一个整数变量 size
。然后调用 initialize
函数初始化矩阵,将矩阵的维度设置为4,并将对角元素分别设置为1、2、3和4。最后使用 printMatrix
函数打印矩阵。
#include
#define MAX_SIZE 100
// 定义对角矩阵结构体
typedef struct {
int size; // 矩阵的维度
int diagonal[MAX_SIZE]; // 存储对角元素的数组
} DiagonalMatrix;
// 初始化对角矩阵
void initialize(DiagonalMatrix *matrix, int size) {
matrix->size = size;
// 初始化对角元素数组
for (int i = 0; i < size; i++) {
matrix->diagonal[i] = 0;
}
}
// 设置对角矩阵中指定位置的元素值
void setElement(DiagonalMatrix *matrix, int row, int col, int value) {
if (row != col) {
printf("Error: Only diagonal elements can be set.\n");
} else if (row < 0 || row >= matrix->size || col < 0 || col >= matrix->size) {
printf("Error: Invalid row or column index.\n");
} else {
matrix->diagonal[row] = value;
}
}
// 获取对角矩阵中指定位置的元素值
int getElement(DiagonalMatrix *matrix, int row, int col) {
if (row < 0 || row >= matrix->size || col < 0 || col >= matrix->size) {
printf("Error: Invalid row or column index.\n");
return 0;
} else if (row != col) {
return 0;
} else {
return matrix->diagonal[row];
}
}
// 打印对角矩阵
void printMatrix(DiagonalMatrix *matrix) {
for (int i = 0; i < matrix->size; i++) {
for (int j = 0; j < matrix->size; j++) {
printf("%d ", getElement(matrix,i,j));
}
printf("\n");
}
}
int main() {
DiagonalMatrix matrix;
int size = 6;
initialize(&matrix, size);
// 读入数据并设置对角矩阵的元素
for (int i = 0; i < size; i++) {
int value;
printf("\nEnter the value for element [%d][%d]: ", i, i);
scanf("%d", &value);
setElement(&matrix, i, i, value);
}
printf("Diagonal Matrix:\n");
printMatrix(&matrix);
return 0;
}