矩阵 AAA 规模为 n×mn\times mn×m,矩阵 BBB 规模为 m×pm\times pm×p,现需要你求 A×BA\times BA×B。
矩阵相乘的定义:n×mn\times mn×m 的矩阵与 m×pm\times pm×p 的矩阵相乘变成 n×pn\times pn×p 的矩阵,令 aika_{ik}aik 为矩阵 AAA 中的元素,bkjb_{kj}bkj 为矩阵 BBB 中的元素,则相乘所得矩阵 CCC 中的元素
cij=∑k=1maikbkjc_{ij}=\sum_{k=1}^m a_{ik}b_{kj}cij=k=1∑maikbkj
具体可见样例。
输入格式
第一行两个数 n,mn,mn,m;
接下来 nnn 行 mmm 列描述一个矩阵 AAA;
接下来一行输入 ppp;
接下来 mmm 行 ppp 列描述一个矩阵 BBB。
输出格式
输出矩阵 AAA 与矩阵 BBB 相乘所得的矩阵 CCC。
样例
Input | Output |
---|---|
2 3 1 2 3 3 2 1 2 1 1 2 2 3 3 | 14 14 10 10 |
[14=1×1+2×2+3×314=1×1+2×2+3×310=3×1+2×2+1×310=3×1+2×2+1×3] \begin{bmatrix} 14=1\times 1+2\times 2+3\times 3&14=1\times 1+2\times 2+3\times 3\ 10=3\times 1+2\times 2+1\times 3&10=3\times 1+2\times 2+1\times 3 \end{bmatrix} [14=1×1+2×2+3×310=3×1+2×2+1×314=1×1+2×2+3×310=3×1+2×2+1×3]
数据范围与提示
对于全部数据,1≤n,m,p≤100,−10000≤aij,bij≤100001\le n,m,p \le 100,-10000\le a_{ij},b_{ij}\le 100001≤n,m,p≤100,−10000≤aij,bij≤10000。
#include<bits/stdc++.h>
using namespace std;
int n,m,p;
int a[105][105],b[105][105],c[105][105];
int main()
{
cin>>n>>m;
for(int i=0; i<n; i++)
for(int j=0; j<m; j++)
cin>>a[i][j];
cin>>p;
for(int i=0; i<m; i++)
for(int j=0; j<p; j++)
cin>>b[i][j];
memset(c,0,sizeof(c));
for(int i=0; i<n; i++)
for(int j=0; j<m; j++)
for(int k=0; k<p; k++)
c[i][k]+=a[i][j]*b[j][k];
for(int i=0; i<n; i++)
{
cout<<c[i][0];
for(int j=1; j<p; j++)
cout<<" "<<c[i][j];
cout<<endl;
}
return 0;
}