矩阵的乘积【水题】

给定两个矩阵$A$和$B$，你需要判断它们是否可以相乘，若可以请输出相乘后的矩阵。

Input

第一行输入一个整数$T$，代表有$T$组测试数据。

每组数据第一行输入两个整数$N,M$，代表矩阵$A$的行、列。

接下来$N$行，每行输入$M$个整数$a[][]$。

之后一行输入两个整数$n,m$，代表矩阵$B$的行、列。

接下来$n$行，每行输入$m$个整数$b[][]$。

注：$1 <= T <= 500，1 <= N,m <= 100，1 <= M,n <= 10，1 <= $矩阵元素$<= 20$。

Output

若矩阵$A、B$可以相乘，先输出$YES$，再输出相乘得到的矩阵。

对每行的矩阵元素，每两个元素之间有一个空格，最后一个没有空格。

反之输出$NO$。

Sample Input

2
2 2
1 1
1 1
2 3
1 1 1
1 1 1
2 3
1 1 1
1 1 1
2 2
1 1
1 1 

Sample Output

YES
2 2 2
2 2 2
NO 

AC代码：
#include<cstdio>
int a[200][201],b[201][201],c[201][201];
int main()
{
	int t,i,j,k,n,m,N,M;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&m);
		for(i=0; i<n; i++)
		{
			for(j=0; j<m; j++)
			scanf("%d",&a[i][j]);
		}
         scanf("%d%d",&N,&M);
         for(i=0; i<N; i++)
         {
         	for(j=0; j<M; j++)
         	scanf("%d",&b[i][j]);
		 }
		 if(m==N)
		 {
		 	printf("YES\n");
		 	for(i=0; i<n; i++)
		 	{
		 		for(j=0; j<M; j++)
		 		{
		 			c[i][j]=0;  //注意要初始化 
		 			for(k=0; k<m; k++)
		 			{
					 c[i][j]+=a[i][k]*b[k][j];
				}
				printf("%d%c",c[i][j],j==M-1?'\n':' ');
				 }
			 }
		}
	/*	 for(i=0; i<n; i++)
			 {
			 	for(j=0; j<M; j++)
			 {
				 	printf("%d ",c[i][j]);
				 }
				 printf("\n");
			 }
		 	
		 }*/
		 else
		 printf("NO\n");
	}
	return 0;
}