蓝桥杯算法训练 矩阵乘法

题目链接

问题描述
  输入两个矩阵,分别是ms,sn大小。输出两个矩阵相乘的结果。
输入格式
  第一行,空格隔开的三个正整数m,s,n(均不超过200)。
  接下来m行,每行s个空格隔开的整数,表示矩阵A(i,j)。
  接下来s行,每行n个空格隔开的整数,表示矩阵B(i,j)。
输出格式
  m行,每行n个空格隔开的整数,输出相乘後的矩阵C(i,j)的值。
样例输入
2 3 2
1 0 -1
1 1 -3
0 3
1 2
3 1
样例输出
-3 2
-8 2

提示
矩阵C应该是m行n列,其中C(i,j)等于矩阵A第i行行向量与矩阵B第j列列向量的内积。
例如样例中C(1,1)=(1,0,-1) * (0,1,3) = 1 * 0 +0*1+(-1)*3=-3

代码:

#include<bits/stdc++.h>
using namespace std;
int A[202][202], B[202][202], C[202][202];
int main()
{
	ios::sync_with_stdio(false);
	int m, s, n;
	while(cin >> m >> s >> n)
	{
		memset(A, 0, sizeof(A));
		memset(B, 0, sizeof(B));
		memset(C, 0, sizeof(C));
		for(int i = 0; i < m; i++)		//4e4
			for(int j = 0; j < s; j++)
				cin >> A[i][j];
		
		for(int i = 0; i < s; i++)		//4e4
			for(int j = 0; j < n; j++)
				cin >> B[i][j];
		
		for(int i = 0; i < m; i++)
			for(int j = 0; j < n; j++)	//8e6
				for(int k = 0; k < s; k++)
					C[i][j] += (A[i][k] * B[k][j]);
		
		for(int i = 0; i < m; i++)
		{
			for(int j = 0; j < n; j++)
				if(j == 0) cout << C[i][j];
				else cout << " " << C[i][j];
			cout << endl;
		}
	}
	return 0;
}
上一篇:Caocao's Bridges hdu-4738


下一篇:2021-01-09