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; }

View Code

 題解效率