矩阵乘法(矩阵)【hpu】

点击打开链接

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

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 

代码:

#include <cstdio>
#include <algorithm>
using namespace std;
int a[105][105],b[105][105],c[105][105];
int main()
{
	int t;
	int N,M;
	int n,m;
	scanf("%d",&t);
	while(t--)
	{	
		scanf("%d %d",&N,&M);
		for(int i=0;i<N;i++)
		   for(int j=0;j<M;j++)
		scanf("%d",&a[i][j]);
		
		scanf("%d %d",&n,&m);
		for(int i=0;i<n;i++)
		   for(int j=0;j<m;j++)
		scanf("%d",&b[i][j]);
		if( M!=n )
       	printf("NO\n");
       	else
		{
			printf("YES\n");
				for(int i=0;i<N;i++)
	               for(int k=0;k<m;k++)
	           {
	           	      c[i][k]=0;        //注意,wa了好几次全因为这,艹 
		               for(int j=0;j<M;j++)
		           {
		              	c[i][k] += a[i][j]*b[j][k];
	               }
	           }
	           for(int i=0;i<N;i++)
              {
              	for(int j=0;j<m;j++)
	           	printf("%d%c",c[i][j],j==m-1?'\n':' ');
			  }
       	}

	}
	return 0;
}


猜你喜欢

转载自blog.csdn.net/hpuw1234/article/details/77162653
今日推荐