矩阵乘法
题目描述
计算两个矩阵的乘法。 n × m n \times m n×m 阶的矩阵 A A A 乘以 m × k m \times k m×k 阶的矩阵 B B B 得到的矩阵 C C C 是 n × k n \times k n×k 阶的,且 C [ i ] [ j ] = A [ i ] [ 0 ] × B [ 0 ] [ j ] + A [ i ] [ 1 ] × B [ 1 ] [ j ] + C[i][j]=A[i][0] \times B[0][j]+A[i][1] \times B[1][j]+ C[i][j]=A[i][0]×B[0][j]+A[i][1]×B[1][j]+ …… + A [ i ] [ m − 1 ] × B [ m − 1 ] [ j ] ( C [ i ] [ j ] +A[i][m-1] \times B[m-1][j](C[i][j] +A[i][m−1]×B[m−1][j](C[i][j] 表示 C C C 矩阵中第 i i i 行第 j j j 列元素)。
输入格式
第一行为 n , m , k n,m,k n,m,k,表示 A A A 矩阵是 n n n 行 m m m 列, B B B 矩阵是 m m m 行 k k k 列, n , m , k n,m,k n,m,k 均小于 100 100 100。
然后先后输入 A A A 和 B B B 两个矩阵, A A A 矩阵 n n n 行 m m m 列, B B B 矩阵 m m m 行 k k k 列,矩阵中每个元素的绝对值不会大于 1000 1000 1000。
输出格式
输出矩阵 C C C,一共 n n n 行,每行 k k k 个整数,整数之间以一个空格分开。
样例 #1
样例输入 #1
3 2 3
1 1
1 1
1 1
1 1 1
1 1 1
样例输出 #1
2 2 2
2 2 2
2 2 2
题解
#include <bits/stdc++.h>
using namespace std;
int a[110][110], b[110][110], c[110][110];
int n, m, k;
int main() {
// 读取矩阵的行数和列数
cin >> n >> m >> k;
// 读取矩阵A
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
cin >> a[i][j];
// 读取矩阵B
for (int i = 1; i <= m; i++)
for (int j = 1; j <= k; j++)
cin >> b[i][j];
// 计算矩阵C
for (int i = 1; i <= n; i++)
for (int j = 1; j <= k; j++)
for (int l = 1; l <= m; l++)
c[i][j] += a[i][l] * b[l][j]; // 矩阵乘法
// 输出结果矩阵C
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= k; j++)
cout << c[i][j] << " ";
cout << endl;
}
return 0;
}
题目来源
洛谷