天梯赛训练2 7-8 矩阵A乘以B

知道矩阵A*B的规则按照规则来做就好了

#include <bits/stdc++.h>
#include <stdio.h>
using namespace std;
int a[1005][1005],b[1005][1005];
int main()
{
   ios::sync_with_stdio(false);
   int N1,M1,N2,M2,i,j;
   cin >> N1>>M1;
   for(i=1;i<=N1;i++)
    for(j=1;j<=M1;j++)
      cin >> a[i][j];
    cin >> N2 >> M2;
   for(i=1;i<=N2;i++)
    for(j=1;j<=M2;j++)
      cin >> b[i][j];
   if(M1!=N2)
     printf("Error: %d != %d\n",M1,N2);
   else
   {
       int k=1,i=1,sum;
       cout << N1 <<" "<< M2 <<endl;
       for(i=1;i<=N1;i++)
       {
           for(k=1;k<=M2;k++)
           {
               sum=0;
               for(j=1;j<=N2;j++)
               {
                   sum+=a[i][j]*b[j][k];
               }
               if(k<M2) cout << sum <<" ";
               else cout <<sum <<endl;
           }
       }
   }
   return 0;
}