hpu暑假训练 B 矩阵乘法【矩阵】

B - 矩阵乘法

Time Limit:1000MS     Memory Limit:131072KB     64bit IO Format:%lld & %llu

Submit  Status

Description

给定两个矩阵$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 


解析：
   注意数组的初始化，博主就因为这个WA了一下。还有就是矩阵可以相乘的条件就是
   矩阵A的列数等于矩阵B的行数。

程序如下：
#include<cstdio>
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int n,m,p,q;
		int a[103][103],b[103][103],c[103][103]={0};
		scanf("%d%d",&m,&n);
		int i,j,k;
		for(i=0; i<m; i++)
		    for(j=0; j<n; j++)
		        scanf("%d",&a[i][j]);
		scanf("%d%d",&p,&q);
		for(i=0; i<p; i++)
		    for(j=0; j<q; j++)
		        scanf("%d",&b[i][j]);
		if(n==p)
		{
			printf("YES\n");
			for(i=0; i<m; i++)
			  for(j=0; j<n; j++)
				for(k=0; k<q; k++)        
				    c[i][k]+=a[i][j]*b[j][k];
			for(i=0; i<m; i++)
				for(k=0; k<q; k++)
			        printf("%d%c",c[i][k],k==q-1?'\n':' ');
		}
		else
		    printf("NO\n");
	}
	return 0;
}