HOU VJ【1035】矩阵乘法

矩阵乘法

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 

Hint

Source

CZY

几天本来写了两道乘法题，但其实原理都一样就是输入输出格式不同，第二道题得再加一个二维数组才能过，没看清题，所以错了几次，注意这一点，除了这个 其他基本没啥区别了，代码稍微改下格式就过了：

第二道题的链接 想看的话可以看下

下面就是这道题的AC代码了：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[1000][1000],b[1000][1000];
int i,j,p; 
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
	int m,s,n,q;
	scanf("%d %d",&m,&s);
	for(i=0;i<m;i++)
	{
		for(j=0;j<s;j++)
		scanf("%d",&a[i][j]);
	}
	scanf("%d %d",&n,&q);
	for(i=0;i<n;i++)
	{
		for(j=0;j<q;j++)
		scanf("%d",&b[i][j]);
	}
	if(s!=n)
	{
		printf("NO\n");
		continue;
	}
	else
	{
	printf("YES\n");
	for(i=0;i<m;i++)
	{	
		for(p=0;p<q;p++)
		{
		int sum=0;
		for(j=0;j<n;j++)
		{
			sum=sum+a[i][j]*b[j][p];
	    }
        printf("%d",sum);
        if(p!=q-1)
        printf(" ");
	    }
		printf("\n");  
	}
	}
   }
	return 0;
}