蓝桥杯 算法提高 矩阵相乘

时间:2023-02-13 18:09:07

算法提高 矩阵相乘
时间限制:1.0s 内存限制:256.0MB
提交此题
问题描述
  小明最近在为线性代数而头疼,线性代数确实很抽象(也很无聊),可惜他的老师正在讲这矩阵乘法这一段内容。
  当然,小明上课打瞌睡也没问题,但线性代数的习题可是很可怕的。
  小明希望你来帮他完成这个任务。

  现在给你一个ai行aj列的矩阵和一个bi行bj列的矩阵,
  要你求出他们相乘的积(当然也是矩阵)。
  (输入数据保证aj=bi,不需要判断)
输入格式
  输入文件共有ai+bi+2行,并且输入的所有数为整数(long long范围内)。
  第1行:ai 和 aj
  第2~ai+2行:矩阵a的所有元素
  第ai+3行:bi 和 bj
  第ai+3~ai+bi+3行:矩阵b的所有元素
输出格式
  输出矩阵a和矩阵b的积(矩阵c)
  (ai行bj列)
样例输入
2 2
12 23
45 56
2 2
78 89
45 56
样例输出
1971 2356
6030 7141

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
#include <stdlib.h>
#include <map>
using namespace std;
long long a[105][105];//矩阵1
long long b[105][105];//矩阵2
long long c[105][105];//结果矩阵
void in(long long n,long long m,long long (&x)[105][105])//函数调用数组输入
{
for(long long i=0;i<n;i++)
for(long long j=0;j<m;j++)
cin>>x[i][j];
}
void sout(long long n,long long m,long long l)//矩阵相乘并输出
{
memset(c,0,sizeof(c));

for(long long i=0;i<n;i++)
{
for(long long k=0;k<l;k++)
for(long long j=0;j<m;j++)
{
c[i][k]+=a[i][j]*b[j][k];
}
}

for(long long i=0;i<n;i++)
{
for(long long j=0;j<l;j++)
{
cout<<c[i][j]<<' ';
}
cout<<endl;
}
}
int main()
{
long long n,m,l;
while(cin>>n>>m)
{
in(n,m,a);
cin>>m>>l;
in(m,l,b);
sout(n,m,l);
}
}