蓝桥杯——算法提高 矩阵相乘

算法提高 矩阵相乘  

时间限制：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<stdio.h>

#include <stdlib.h>

int main()

{

 

long long **a,**b,**c;

int a1,a2,b1,b2,i,j,k;

scanf("%d%d",&a1,&a2);//二维数组的空间申请方法

a=(long long**)malloc(sizeof(long long*)*a1);  

    for(i=0;i<a1;i++)  

    a[i]=(long long*)malloc(sizeof(long long)*a2);

for(i=0;i<a1;i++)

for(j=0;j<a2;j++)

scanf("%lld",&a[i][j]);

scanf("%d%d",&b1,&b2);

b=(long long**)malloc(sizeof(long long*)*b1);  

    for(i=0;i<b1;i++)  

    b[i]=(long long*)malloc(sizeof(long long)*b2);

for(i=0;i<b1;i++)

for(j=0;j<b2;j++)

scanf("%lld",&b[i][j]);

 

c=(long long**)malloc(sizeof(long long*)*a1);  

    for(i=0;i<a1;i++)  

    c[i]=(long long*)malloc(sizeof(long long)*b2);

 

for(i=0;i<a1;i++)

for(j=0;j<b2;j++)

c[i][j]=0;

 

    for(i=0;i<a1;i++)

for(j=0;j<b2;j++)

{

for(k=0;k<a2;k++)

c[i][j]+=a[i][k]*b[k][j];

}

 

for(i=0;i<a1;i++)

for(j=0;j<b2;j++)

{

printf("%lld ",c[i][j]);

if(j==b2-1)

printf("\n");

}

 

for(i=0;i<a1;i++)

free(a[i]);

free(a);

for(i=0;i<b1;i++)

free(b[i]);

free(b);

return 0;

}