算法训练 矩阵乘法

问题描述
　　输入两个矩阵，分别是m*s，s*n大小。输出两个矩阵相乘的结果。
输入格式
　　第一行，空格隔开的三个正整数m,s,n（均不超过200）。
　　接下来m行，每行s个空格隔开的整数，表示矩阵A（i，j）。
　　接下来s行，每行n个空格隔开的整数，表示矩阵B（i，j）。
输出格式
　　m行，每行n个空格隔开的整数，输出相乘後的矩阵C（i，j）的值。
样例输入
2 3 2
1 0 -1
1 1 -3
0 3
1 2
3 1
样例输出
-3 2
-8 2

提示
矩阵C应该是m行n列，其中C(i,j)等于矩阵A第i行行向量与矩阵B第j列列向量的内积。
例如样例中C(1,1)=(1,0,-1)*(0,1,3) = 1 * 0 +0*1+(-1)*3=-3

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
int a[10000];
int b[10000];
using namespace std;
int main(){
int m,s,n; //m*s，s*n
int i,j,k,s_t,t,aa;
int sum[1000]={0};

scanf("%d %d %d",&m,&s,&n);
int tot=1;
for(i=1;i<=m*s;i++){
scanf("%d",&a[tot++]);

}
tot=1;
for(i=1;i<=s*n;i++){
scanf("%d",&b[tot++]);
}

k=1;i=1,s_t=1;
while(k<=m){
memset(sum,0,sizeof(sum));
for(j=1;j<=s*n;i++,j+=n){
for(aa=0;aa<n;aa++)
sum[aa]+=a[i]*b[j+aa];
}
for(t=0;t<n;t++)
printf("%d ",sum[t]);
puts("");
k++;
}

return 0;
}最后一个测试用例 运行超时！原因是使用memset(sum,0,sizeof(sum));  每次循环都会把数组元素置零。



下面是ac的代码（别人家的）#include<stdio.h>
int main()
{
int m,s,n,a[200][200],b[200][200],c[200][200];
int i,j,k,q;
scanf("%d%d%d",&m,&s,&n);
for(i=0;i<m;i++)
for(j=0;j<s;j++)
scanf("%d",&a[i][j]);
for(i=0;i<s;i++)
for(j=0;j<n;j++)
scanf("%d",&b[i][j]);
for(i=0;i<m;i++)
{
for(k=0;k<n;k++)
{
for(j=0;j<s;j++)
{
c[i][k]+=a[i][j]*b[j][k]; //i代表矩阵a的行，j是矩阵b的行，k是矩阵b的列
}
}
}
for(i=0;i<m;i++)
{
for(k=0;k<n;k++)
{
printf("%d",c[i][k]);
if(k==n-1)
printf("\n");
else
printf(" ");
}

}
return 0;

}