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;
}