问题描述
小明最近刚刚学习了矩阵乘法,但是他计算的速度太慢,于是他希望你能帮他写一个矩阵乘法的运算器。
输入格式
输入的第一行包含三个正整数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<N,M,K<=100, 0<=矩阵中的每个数<=1000。
要点
1.参考了柳神的做法,用动态数组来表示一个数组:
vector< vector<int> > a(m, vector<int>(s));//数组a,m行s列
这行代码表示:创造一个m行s列的a数组。
具体而言,也就是:
- 定义了一个vector容器a,元素类型为vector <int>,
- 初始化为包含m个vector <int>对象,每个对象都是一个新创立的vector对象的拷贝,而这个新创立的vector <int>对象被初始化为包含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;
}