【华为机试】HJ69 矩阵乘法

知识点与难度

数组、中等难度

描述

如果A是个x行y列的矩阵，B是个y行z列的矩阵，把A和B相乘，其结果将是另一个x行z列的矩阵C。这个矩阵的每个元素是由下面的公式决定的

矩阵的大小不超过100*100

输入描述：

第一行包含一个正整数x，代表第一个矩阵的行数
第二行包含一个正整数y，代表第一个矩阵的列数和第二个矩阵的行数
第三行包含一个正整数z，代表第二个矩阵的列数
之后x行，每行y个整数，代表第一个矩阵的值
之后y行，每行z个整数，代表第二个矩阵的值

输出描述：

对于每组输入数据，输出x行，每行z个整数，代表两个矩阵相乘的结果

输入输出示例：

输入：

2
3
2
1 2 3
3 2 1
1 2
2 1
3 3

输出：

14 13
10 11

我的代码：

#include <iostream>
using namespace std;

int main() {
    int x, y, z;
    cin >> x;
    cin >> y;
    cin >> z;
    int sum = 0;
    int AM[100][100] = {0};
    int BM[100][100] = {0};

    for(int i=0;i<x;i++){
        for(int j=0;j<y;j++){
            cin >> AM[i][j];
        }
    }

    for(int i=0;i<y;i++){
        for(int j=0;j<z;j++){
            cin >> BM[i][j];
        }
    }


    for(int i=0;i<x;i++){
        for(int j=0;j<z;j++){
            sum = 0;
            for(int k=0;k<y;k++){
                //cout<<AM[i][k]<<'*'<<BM[k][j]<<endl;
                sum = sum + AM[i][k]*BM[k][j];
            }
            cout<<sum<<" ";
        }
        cout<<endl;
    }


}
// 64 位输出请用 printf("%lld")

我的代码分析：

简单粗暴的三重循环。

top1代码：

#include<iostream>
#include<vector>
using namespace std;
int main(){
    int x, y, z;
    while (cin >> x >> y >> z){
        vector<vector<int>> arr1(x, vector<int>(y, 0));
        vector<vector<int>> arr2(y, vector<int>(z, 0));
        vector<vector<int>> arr3(x, vector<int>(z, 0));
        for(int i = 0; i < x; ++i){
            for(int j = 0; j < y; ++j)
                cin >> arr1[i][j];
        }
        for(int i = 0; i < y; ++i){
            for(int j = 0; j < z; ++j)
                cin >> arr2[i][j];
        }
        for(int i = 0; i < x; ++i){
            for(int j = 0; j < y; ++j)
                for(int k = 0; k < z; ++k)
                    arr3[i][k] += arr1[i][j] * arr2[j][k];
        }
        for(int i = 0; i < x; ++i){
            for(int j = 0; j < z-1; ++j)
                cout << arr3[i][j] << " ";
            cout << arr3[i][z-1] << endl;
        }
    }
    return 0;
}

top1代码分析：

思路好像也差不多啊，但是他这个运行时间就比我少好多，难道是vector的原因吗？

总结：

以后学习学习vector的用法。