DAY eight 二维数组和矩阵和压缩矩阵转置

矩阵乘法总代码

#include<stdio.h>
#include<stdlib.h>
 
/**
 *动态二维数组
 */
 
typedef struct TwoDArray{
    int rows;
    int columns;
    int **elements;
} TwoDArray, *TwoDArrayPtr;
 
/**
 *静态二维数组
 */
 
typedef struct TwoDStaticArray{
    int rows;
    int columns;
    int elements[4][5];
} TwoDStaticArray, *TwoDStaticArrayPtr;
 
/**
 *创建数组
 */
 
TwoDArrayPtr initTwoDArray(int paraRows, int paraColumns) {
    int i;
    TwoDArrayPtr resultPtr = (TwoDArrayPtr)malloc(sizeof(TwoDArray));
    resultPtr->rows = paraRows;
    resultPtr->columns = paraColumns;
    resultPtr->elements = (int **)malloc(sizeof(int *) * paraRows);
    for (i = 0; i < paraRows; i ++) {
        resultPtr->elements[i] = (int *)malloc(sizeof(int) * paraColumns);
    }
    
    return resultPtr;
}
 
/**
 *赋值
 */
 
void randomizeTwoDArray(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("\n");
    }
}
 
/**
 *矩阵相乘
 */
 
TwoDArrayPtr matrixMultiply(TwoDArrayPtr paraPtr1, TwoDArrayPtr paraPtr2) {
    int i, j, k, sum;
    if (paraPtr1->columns != paraPtr2->rows) {
        printf("Matrices cannot be multiplied.\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);
        }
        printf("\n");
    }
    
    return resultPtr;
}
 
void twoDArrayTest() {
    TwoDArrayPtr tempPtr1, tempPtr2, tempPtr3;
    tempPtr1 = initTwoDArray(3, 2);
    randomizeTwoDArray(tempPtr1, 1, 5);
    printf("The first matrix:\n");
    printTwoDArray(tempPtr1);
    
    tempPtr2 = initTwoDArray(2, 4);
    randomizeTwoDArray(tempPtr2, 4, 9);
    printf("The second matrix:\n");
    printTwoDArray(tempPtr2);
    
    tempPtr3 = matrixMultiply(tempPtr1, tempPtr2);
    printf("The result:\n");
    printTwoDArray(tempPtr3);
}
 
/**
 *创建静态数组 
 */
 
TwoDStaticArrayPtr initTwoDStaticArray() {
    int i, j;
    TwoDStaticArrayPtr resultPtr = (TwoDStaticArrayPtr)malloc(sizeof(TwoDStaticArray));
    resultPtr->rows = 4;
    resultPtr->columns = 5;
    for (i = 0; i < 4; i ++) {
        for (j = 0; j < 5; j ++) {
            resultPtr->elements[i][j] = i * 10 + j;
            printf("(%d, %d): %p\n", i, j, &(resultPtr->elements[i][j]));
        }
    }
    
    return resultPtr;
}
 
int main() {
    twoDArrayTest();
    TwoDStaticArrayPtr tempPtr = initTwoDStaticArray();
    
    return 0;
}

运行结果

转置代码总代码

#include<stdio.h>
#include<stdlib.h>
 
/**
 *创建一个含有行与列及值的结构体
 */
 
typedef struct Tripe{
    int i;
    int j;
    int e;
} Triple, *TriplePtr;
 
typedef struct CompressedMatrix{
    int rows, columns, numElements;
    TriplePtr elements;
} CompressedMatrix, *CompressedMatrixPtr;
 
/**
 *初始化
 */
 
CompressedMatrixPtr initCompressedMatrix(int paraRows, int paraColumns, int paraElements, int **paraData) {
    int i;
    CompressedMatrixPtr resultPtr = (CompressedMatrixPtr)malloc(sizeof(CompressedMatrix));
    resultPtr->rows = paraRows;
    resultPtr->columns = paraColumns;
    resultPtr->numElements = paraElements;
    resultPtr->elements = (TriplePtr)malloc(sizeof(Triple) * paraElements);
    
    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\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(sizeof(int) * paraPtr->columns);
    int *tempOffsets = (int *)malloc(sizeof(int) * paraPtr->columns);
    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(sizeof(Triple) * paraPtr->numElements);
    
    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\n", i, tempOffsets[i]);
    }
    
    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;
}
 
void compressedMatrixTest() {
    CompressedMatrixPtr tempPtr1, tempPtr2;
    int i, j, tempElements;
    
    tempElements = 4;
    int ** tempMatrix1 = (int **)malloc(sizeof(int *) * tempElements);
    for (i = 0; i < tempElements; i ++) {
        tempMatrix1[i] = (int *)malloc(sizeof(int) * 3);
    }
    
    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\n");
    printCompressedMatrix(tempPtr1);
    
    tempPtr2 = transposeCompressedMatrix(tempPtr1);
    printf("After transpose.\n");
    printCompressedMatrix(tempPtr2);
}
 
int main() {
    compressedMatrixTest();
    
    return 0;
}

运行结果