信息学奥赛一本通(c++):1125:矩阵乘法
一、题目
1125:矩阵乘法
时间限制: 1000 ms 内存限制: 65536 KB
【题目描述】
计算两个矩阵的乘法。n×m阶的矩阵A乘以m×k阶的矩阵B得到的矩阵C 是n×k阶的,且C[i][j] = A[i][0]×B[0][j] + A[i][1]×B[1][j] + …… +A[i][m-1]×B[m-1][j](C[i][j]表示C矩阵中第i行第j列元素)。
【输入】
第一行为n, m, k,表示A矩阵是n行m列,B矩阵是m行k列,n, m, k均小于100。
然后先后输入A和B两个矩阵,A矩阵n行m列,B矩阵m行k列,矩阵中每个元素的绝对值不会大于1000。
【输出】
输出矩阵C,一共n行,每行k个整数,整数之间以一个空格分开。
【输入样例】
3 2 3 1 1 1 1 1 1 1 1 1 1 1 1
【输出样例】
2 2 2 2 2 2 2 2 2
二、思路
矩阵乘法大学时候有门课叫《线性代数》里会学到,简单来说就是把一个矩阵 第一行的每一个元素与第二个矩阵 第一列的每一个元素 对应相乘 然后全部加起来 作为新矩阵的第一行第一列的一个元素
1.输入两个矩阵
2.三层for循环 矩阵乘法 第一层是第一个矩阵的行 第二层是第二个矩阵的列 第三层是第一个的列或者第二个的行
三、代码
#include<iostream>
using namespace std;
int main()
{
int n,m,k;
cin>>n>>m>>k;
int a[n][m],b[m][k],c[n][k];
for(int i=0; i<n; i++)
{
for(int j=0; j<m; j++)
{
cin>>a[i][j];
}
}
for(int i=0; i<m; i++)
{
for(int j=0; j<k; j++)
{
cin>>b[i][j];
}
}
for(int i=0; i<n; i++)
{
for(int j=0; j<k; j++)
{
c[i][j]=0;
for(int l=0; l<m; l++)
{
c[i][j]+=a[i][l]*b[l][j];
}
}
}
for(int i=0; i<n; i++)
{
for(int j=0; j<k; j++)
{
cout<<c[i][j]<<" ";
}
cout<<endl;
}
return 0;
}