矩阵连乘-结构体与动态分配数组

 在这里我使用的方法主要是两个：使用结构体Matrix、为数组动态分配内存。

由于数组不能直接使用变量赋值，所以动态分配数组时很有用的方法！

将在结构体中，我保存了该矩阵的行数、列数，还定义了一个int**的变量用来存储数组数据，方便动态分配内存。

//“矩阵”结构体
struct Matrix {
    int rowNum;
    int colNum;
    int** data;
};

动态分配二重数组的方法如下：

//对于 int ** data 动态分配内存
int** data = new int* [rowNum];

for (int i = 0; i < rowNum; i++){
    data[j] = new int[colNum];
} 

//将数据填入data
for (int j = 0; j < rowNum; j++){
    for (int k = 0; k < colNum; k++){
        cin >> data[j][k];
    }
} 

总的代码如下，有兴趣可以看一看。

#include <iostream>
using namespace std;

//“矩阵”结构体
struct Matrix {
    int rowNum;
    int colNum;
    int** data;
};

//初始化 mats[]
void initialize_mats(int n, Matrix* mats) {
    for (int i = 0; i < n; i++) {
        int rowNum, colNum;
        cin >> rowNum >> colNum;
        mats[i].rowNum = rowNum;
        mats[i].colNum = colNum;

        mats[i].data = new int* [rowNum];
        for (int j = 0; j < rowNum; j++){
            mats[i].data[j] = new int[colNum];
        } //对于 int ** data 动态分配内存

        for (int j = 0; j < rowNum; j++){
            for (int k = 0; k < colNum; k++){
                cin >> mats[i].data[j][k];
            }
        } //将数据填入data
    } 
}

// 初始化 anws[]
void initialize_anws(int n, Matrix* mats, Matrix* anws) {
    for (int i = 0; i < n; i++) {
        anws[i].rowNum = mats[0].rowNum;
        anws[i].colNum = mats[i].colNum;

        anws[i].data = new int* [anws[i].rowNum];
        for (int j = 0; j < anws[i].rowNum; j++) {
            anws[i].data[j] = new int[anws[i].colNum];
        } //动态分配内存
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < anws[i].rowNum; j++) {
            for (int k = 0; k < anws[i].colNum; k++) {
                anws[i].data[j][k] = 0;
            }
        } //将0填满data
    }

    for (int j = 0; j < anws[0].rowNum; j++) {
        for (int k = 0; k < anws[0].colNum; k++) {
            anws[0].data[j][k] = mats[0].data[j][k];
        }
    } // 将anws[0]赋值为mats[0]
}

//打印矩阵
void printMat(Matrix* mat) {
    for (int i = 0; i < mat->rowNum; i++)
    {
        for (int j = 0; j < mat->colNum; j++)
        {
            cout << mat->data[i][j] << " ";
        }
        cout << endl;
    }
}

//实现矩阵乘法
void Matrix_mul(Matrix* mats, Matrix* anws, int i) { // i为第i次做乘法
    for (int j = 0; j < anws[i].rowNum; j++){ //第j行
        for (int k = 0; k < anws[i].colNum; k++){ //第k列
            for (int n = 0; n < mats[i].rowNum; n++){ //第n个元素
                anws[i].data[j][k] = anws[i].data[j][k] + anws[i - 1].data[j][n] * mats[i].data[n][k];
            }
        }
    }
}

void go() {
    int n;
    cin >> n;
    Matrix* mats = new Matrix[n];
    Matrix* anws = new Matrix[n];//动态分配内存（放到堆中）
    initialize_mats(n, mats);
    initialize_anws(n, mats, anws);
    //数据录入、数组初始化

    for (int i = 1; i < n; i++){
        Matrix_mul(mats, anws, i);
    }

    printMat(&anws[n - 1]);
}

int main() {
    go();
    return 0;
}