LibreOJ #100.矩阵乘法

题目描述

这是一道模板题。

分别给定 n×p  p×m 的两个矩阵 A 和  B，求 A×B 。

输入格式

第一行三个正整数 n p m ，表示矩阵的长宽。
之后的 n 行，每行 p  个整数，表示矩阵 A 。
之后的 p  行，每行 m  个整数，表示矩阵 B 。

输出格式

输出 n 行，每行 m 个整数，表示矩阵 A×B ，每个数模  10 ^ 9 + 7  输出。

样例

样例输入

3 4 5
-2 -8 -9 8
-10 0 6 -8
-10 -6 6 9
4 -7 5 -5 9
10 -2 -10 5 5
-3 -7 -3 8 -2
-6 7 7 3 -2

样例输出

999999898 149 153 999999929 999999951
999999997 999999979 999999883 74 999999921
999999835 103 55 95 999999857

数据范围与提示

1≤n,p,m≤500, −109≤Ai,j,Bi,j≤109

题目分析

注意题目里面的数全是正整数，故最终答案也是正整数

AC代码

#include<iostream>
using namespace std;
const long long MOD = 1e9 + 7;
const int MAXN = (int)5e2 + 10;
long long a[MAXN][MAXN], b[MAXN][MAXN],res[MAXN][MAXN];
int n, p, m;
void solve(){
    for (int i = 0; i < n; i++){

        for (int j = 0; j < m; j++){
            res[i][j] = 0;
            for (int k = 0; k < p; k++)
                res[i][j] += a[i][k] * b[k][j] % MOD;
            long long ans = res[i][j] % MOD;
            if (ans < 0) ans += MOD;
            cout << ans << ' ';
        }
        cout << endl;
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin >> n >> p >> m;
    for (int i = 0; i < n; i++)
    for (int j = 0; j < p; j++)
        cin >> a[i][j];
    for (int i = 0; i < p; i++)
    for (int j = 0; j < m; j++)
        cin >> b[i][j];
    solve();
    return 0;
}

 题解效率