首页 > 技术文章 > 7-54 矩阵A乘以B (15分)

bigageyuan 2020-10-21 16:26 原文

7-54 矩阵A乘以B (15分)
 

给定两个矩阵AAA和BBB,要求你计算它们的乘积矩阵ABABAB。需要注意的是,只有规模匹配的矩阵才可以相乘。即若AAA有RaR_aRa​​行、CaC_aCa​​列,BBB有RbR_bRb​​行、CbC_bCb​​列,则只有CaC_aCa​​与RbR_bRb​​相等时,两个矩阵才能相乘。

输入格式:

输入先后给出两个矩阵AAA和BBB。对于每个矩阵,首先在一行中给出其行数RRR和列数CCC,随后RRR行,每行给出CCC个整数,以1个空格分隔,且行首尾没有多余的空格。输入保证两个矩阵的RRR和CCC都是正数,并且所有整数的绝对值不超过100。

输出格式:

若输入的两个矩阵的规模是匹配的,则按照输入的格式输出乘积矩阵ABABAB,否则输出Error: Ca != Rb,其中CaAAA的列数,RbBBB的行数。

输入样例1:

2 3
1 2 3
4 5 6
3 4
7 8 9 0
-1 -2 -3 -4
5 6 7 8
 

输出样例1:

2 4
20 22 24 16
53 58 63 28
 

输入样例2:

3 2
38 26
43 -5
0 17
3 2
-11 57
99 68
81 72
 

输出样例2:

Error: 2 != 3


#include<stdio.h>
#include<string.h>
int main()
{
 int a_row,a_col;
 int b_row,b_col;
 scanf("%d %d",&a_row,&a_col);
 int a[a_row][a_col];
 int i,j;
 for(i=0;i<a_row;i++)
  for(j=0;j<a_col;j++)
  {
   scanf("%d",&a[i][j]);
   }
 scanf("%d %d",&b_row,&b_col);
  int b[b_row][b_col];
 for(i=0;i<b_row;i++)
  for(j=0;j<b_col;j++)
  {
   scanf("%d",&b[i][j]);
   }
 if(a_col!=b_row)
 {
  printf("Error: %d != %d\n",a_col,b_row);
  return 0;
 }
 int c[a_row][b_col];
 memset(c,0,a_row*b_col*sizeof(int));
 int k;
 printf("%d %d\n",a_row,b_col);
 for(i=0;i<a_row;i++)
 {
  for(j=0;j<b_col;j++)
  {
   for(k=0;k<a_col;k++)
   c[i][j]+=a[i][k]*b[k][j];
   if(j!=b_col-1)
   printf("%d ",c[i][j]);
   else
   printf("%d\n",c[i][j]);
      }
  
 }
 return 0;
 }
 

推荐阅读