2017年天梯赛全国总决赛题集 L1-8 矩阵A乘以B

#include<stdio.h>
int arr1[1000][1000];
int arr2[1000][1000];
int arr3[1000000];
void fun(int arr1[][1000],int n1,int m1,int arr2[][1000],int n2,int m2)
{
    int i,j,k;
    int temp=0;
    int sum;
    for(k=0;k<n1;k++)
    {
        for(i=0;i<m2;i++)
        {
            sum=0;
            for(j=0;j<m1;j++)
            {
                sum+=arr2[j][i]*arr1[k][j];
            }
            arr3[temp++]=sum;
        }
    }
   printf("%d %d\n",n1,m2);
   for(i=0;i<temp;i++)
   {
       if((i+1)%m2==0)printf("%d\n",arr3[i]);
       else printf("%d ",arr3[i]);
   }
}
int main()
{
    int n1,m1;
    int i,j;
    scanf("%d%d",&n1,&m1);
    for(i=0;i<n1;i++)
    {
        int m;
        for(j=0;j<m1;j++)
        {
            scanf("%d",&m);
            arr1[i][j]=m;
        }
    }

    int n2,m2;
    scanf("%d%d",&n2,&m2);
    for(i=0;i<n2;i++)
    {
        for(j=0;j<m2;j++)
        {
            scanf("%d",&arr2[i][j]);
        }
    }
    if(m1!=n2)printf("Error: %d != %d\n",m1,n2);
    else fun(arr1,n1,m1,arr2,n2,m2);
    return 0;
}