L1-048 矩阵A乘以B (15分)

L1-048 矩阵A乘以B (15分)

给定两个矩阵 \(A\) 和 \(B\)，要求你计算它们的乘积矩阵 \(AB\)。需要注意的是，只有规模匹配的矩阵才可以相乘。即若 \(A\) 有 \(R_​a\) 行、\(C_​a\) 列，\(B\) 有 \(R_b\) 行、\(C_b\) 列，则只有 \(C_​a\) 与 \(R_​b\) 相等时，两个矩阵才能相乘。

输入格式：

输入先后给出两个矩阵 \(A\) 和 \(B\)。对于每个矩阵，首先在一行中给出其行数 \(R\) 和列数 \(C\)，随后 \(R\) 行，每行给出 \(C\) 个整数，以 \(1\) 个空格分隔，且行首尾没有多余的空格。输入保证两个矩阵的 \(R\) 和 \(C\) 都是正数，并且所有整数的绝对值不超过 \(100\)。

输出格式：

若输入的两个矩阵的规模是匹配的，则按照输入的格式输出乘积矩阵 \(AB\)，否则输出 Error: Ca != Rb，其中 Ca 是 \(A\) 的列数，Rb 是 \(B\) 的行数。

输入样例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<bits/stdc++.h>
using namespace std;
#define maxn 105
int a,b,x,y,s[maxn][maxn],t[maxn][maxn],ans[maxn][maxn];
int main()
{
    cin>>a>>b;
    for(int i=1;i<=a;i++)
        for(int j=1;j<=b;j++)
            cin>>s[i][j];
    cin>>x>>y;
    for(int i=1;i<=x;i++)
        for(int j=1;j<=y;j++)
            cin>>t[i][j];
    if(b!=x)
    {
        cout<<"Error: "<<b<<" != "<<x<<endl;
        return 0;
    }
    cout<<a<<' '<<y<<endl;
    for(int k=1;k<=b;k++)
         for(int i=1;i<=a;i++)
            for(int j=1;j<=y;j++)
                ans[i][j]+=s[i][k]*t[k][j];
    for(int i=1;i<=a;i++)
        for(int j=1;j<=y;j++)
            cout<<ans[i][j]<<" \n"[i==y];
    return 0;
}