蓝桥杯 算法提高 矩阵相乘

来源:互联网 发布:服务器编程书籍 编辑:程序博客网 时间:2024/05/16 11:16

算法提高 矩阵相乘
时间限制: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];//矩阵1long long b[105][105];//矩阵2long 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);    }}
0 0
原创粉丝点击