矩阵乘法

点击打开链接

点击打开链接

1035: 矩阵乘法 [矩阵]

时间限制: 1 Sec  内存限制: 128 MB

提交: 190  解决: 78  统计

题目描述

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

输入

第一行输入一个整数$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$。

输出

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

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

反之输出$NO$。

样例输入

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

样例输出

YES
2 2 2
2 2 2
NO

#include<stdio.h>
int a[100][100];
int b[100][100];
int c[100][100];
int main()
{
	int t,N,M,n,m,i,j,sum=0,k;
	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(k=0;k<N;k++)
				for(i=0;i<m;i++)
				{
					for(j=0;j<M;j++)
					sum=sum+a[k][j]*b[j][i];
					c[k][i]=sum;
					sum=0;
				}
			for(i=0;i<N;i++)
				for(j=0;j<m;j++)
				{
					if(j<m-1)
					printf("%d ",c[i][j]);
					else
					printf("%d\n",c[i][j]);
				}
					
		}
		else
		printf("NO\n");	
	}
	return 0;
}