题解 | #矩阵乘法#

矩阵乘法

https://www.nowcoder.com/practice/ebe941260f8c4210aa8c17e99cbc663b

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int m, n, k;
    cin >> m >> n >> k;
    vector<vector<int>> matrix1(m, vector<int>(n));
    vector<vector<int>> matrix2(n, vector<int>(k));
    vector<vector<long long>> res(m, vector<long long>(k,0));

    for(int i=0; i<m; i++){
        for(int j=0; j<n; j++){
            cin >> matrix1[i][j];
        }
    }
    for(int i=0; i<n; i++){
        for(int j=0; j<k; j++){
            cin >> matrix2[i][j];
        }
    }

    for(int i=0; i<m; i++){
        for(int j=0; j<k; j++){
            int len = matrix1[i].size();
            for(int k=0; k<len; k++){
                res[i][j] += matrix1[i][k] * matrix2[k][j];
            }
        }
    }

    for(int i=0; i<m; i++){
        for(int j=0; j<k; j++){
            cout << res[i][j] << ' ';
        }
        cout << endl;
    }

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

非常基本的题目

全部评论

相关推荐

10-29 15:38
门头沟学院 Java
榕城小榕树:难道你简历里写了配送路径优化算法?
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务