[C++]蓝桥杯 ADV-279.矩阵乘法

问题描述
  小明最近刚刚学习了矩阵乘法,但是他计算的速度太慢,于是他希望你能帮他写一个矩阵乘法的运算器。
输入格式
  输入的第一行包含三个正整数N,M,K,表示一个NM的矩阵乘以一个MK的矩阵。接下来N行,每行M个整数,表示第一个矩阵。再接下来的M行,每行K个整数,表示第二个矩阵。
输出格式
  输出有N行,每行K个整数,表示矩阵乘法的结果。
样例输入
2 1 3
1
2
1 2 3
样例输出
1 2 3
2 4 6
数据规模和约定
  0

要点

1.参考了柳神的做法,用动态数组来表示一个数组:

vector< vector<int> > a(m, vector<int>(s));//数组a,m行s列 

这行代码表示:创造一个m行s列的a数组。
具体而言,也就是:

  • 定义了一个vector容器a,元素类型为vector >,
  • 初始化为包含mvector >对象,每个对象都是一个新创立的vector对象的拷贝,而这个新创立的vector >对象被初始化为包含S个0,
  • 并且此种写法支持数组形式(如:a[i][j])来访问容器a的元素。

2.这道题还应该注意的地方是输出部分:
数组a与数组b相乘,最后的结果放在矩阵c中。
输出矩阵c时,

   for(int i=0;i<m;i++)
   {
   	for(int j=0;j<n;j++)
   	{
   		cout<<c[i][j]<<" ";
   	}
   	cout<<endl;
   } 
  • 要注意每个元素之间要输出空格
  • 每行输完之后还要换行

代码

	#include<iostream>
#include<vector>
using namespace std;
int main(void)
{
	int m,s,n;
	cin>>m>>s>>n;
	vector< vector<int> > a(m, vector<int>(s));//数组a,m行s列 
	vector< vector<int> > b(s, vector<int>(n));//数组b,s行n列
	vector< vector<int> > c(m, vector<int>(n));//数组c,用于存放结果,m行n列
	//输入数组a 
	for(int i=0;i<m;i++)
	{
		for(int j=0;j<s;j++)
		{
			cin>>a[i][j];
		}
	} 
	//输入数组b 
	for(int i=0;i<s;i++)
	{
		for(int j=0;j<n;j++)
		{
			cin>>b[i][j];
		}
	} 
	//计算
	for(int i=0;i<m;i++)
	{
		for(int j=0;j<s;j++)
		{
			for(int k=0;k<n;k++)
			{
				c[i][k]=c[i][k]+a[i][j]*b[j][k];//关键式子 
			}
		}
	} 
	//注意输出空格,每一行结束后要换行 
	for(int i=0;i<m;i++)
	{
		for(int j=0;j<n;j++)
		{
			cout<<c[i][j]<<" ";
		}
		cout<<endl;
	} 
	return 0;
}

转载

你可能感兴趣的:(蓝桥杯)