5.18二维数组与矩阵乘法

#include <stdio.h> #include <malloc.h> #include <stdlib.h> #define ROWS 4 #define COLS 5 /** *动态分配 */ typedef struct TwoDArray{ int rows; int cols; int** data; } TwoDArray, *TwoDArrayPtr; /** *静态分配 */ typedef struct TwoDStaticArray{ int rows; int cols; int data[ROWS][COLS]; } TwoDStaticArray, *TwoDStaticArrayPtr; /** * 初始化动态二维数组 */ TwoDArrayPtr initArray(int r, int c){ int i; TwoDArrayPtr p = (TwoDArrayPtr)malloc(sizeof(struct TwoDArray)); p->rows = r; p->cols = c; p->data = (int**)malloc(r * sizeof(int*)); for (i = 0; i < r; i ++){ p->data[i] = (int*)malloc(c * sizeof(int)); } return p; } /** * 随机填充数组 */ void randomArray(TwoDArrayPtr p, int low, int high){ int i, j; for (i = 0; i < p->rows; i ++){ for (j = 0; j < p->cols; j ++) { p->data[i][j] = rand() % (high - low) + low; } } } /** * 打印二维数组 */ void printArray(TwoDArrayPtr p){ int i, j; for (i = 0; i < p->rows; i ++){ for (j = 0; j < p->cols; j ++) { printf("%d, ", p->data[i][j]); } printf("\r\n"); } } /** * 矩阵乘法 */ TwoDArrayPtr mulArray(TwoDArrayPtr a, TwoDArrayPtr b){ int i, j, k, sum; if (a->cols != b->rows){ printf("矩阵无法相乘\r\n"); return NULL; } TwoDArrayPtr res = initArray(a->rows, b->cols); for (i = 0; i < a->rows; i ++){ for (j = 0; j < b->cols; j ++) { sum = 0; for (k = 0; k < a->cols; k ++) { sum += a->data[i][k] * b->data[k][j]; } res->data[i][j] = sum; printf("sum = %d, ", sum); } } return res; } /** * 测试动态二维数组 */ void testDynamic(){ TwoDArrayPtr a, b, c; a = initArray(3, 2); randomArray(a, 1, 5); printf("第一个矩阵:\r\n"); printArray(a); b = initArray(2, 4); randomArray(b, 4, 9); printf("第二个矩阵:\r\n"); printArray(b); c = mulArray(a, b); printf("结果矩阵:\r\n"); printArray(c); } /** * 初始化静态二维数组 */ TwoDStaticArrayPtr initStatic(){ int i, j; TwoDStaticArrayPtr p = (TwoDStaticArrayPtr)malloc(sizeof(struct TwoDStaticArray)); p->rows = ROWS; p->cols = COLS; for (i = 0; i < ROWS; i ++){ for (j = 0; j < COLS; j ++) { p->data[i][j] = i * 10 + j; printf("(%d, %d): %d; ", i, j, &(p->data[i][j])); } } return p; } /** * 主函数 */ int main(){ testDynamic(); TwoDStaticArrayPtr p = initStatic(); return 1; }