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]<<' ';
        cout<<ans[i][y]<<endl;
    }
    return 0;
}