0
点赞
收藏
分享

微信扫一扫

矩阵 A×B (数论)

ZGtheGreat 2022-01-22 阅读 52
矩阵算法

矩阵 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∑m​aik​bkj​

具体可见样例。

输入格式

第一行两个数 n,mn,mn,m;

接下来 nnn 行 mmm 列描述一个矩阵 AAA;

接下来一行输入 ppp;

接下来 mmm 行 ppp 列描述一个矩阵 BBB。

输出格式

输出矩阵 AAA 与矩阵 BBB 相乘所得的矩阵 CCC。

样例

InputOutput
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×3​14=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;
}
举报

相关推荐

0 条评论