hpu-【1035: 矩阵乘法】

1035: 矩阵乘法

时间限制: 1 Sec   内存限制: 128 MB
提交: 16   解决: 8
[ 提交][ 状态][ 讨论版]

题目描述

给定两个矩阵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
<span style="font-size:18px;">#include<cstdio>
int a[110][110],b[110][110];

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int n,m,n2,m2;
		scanf("%d%d",&n,&m);
		int i,j;
		for(i=0;i<n;++i)
		{
			for(j=0;j<m;++j)
				scanf("%d",&a[i][j]); 
		} 
		scanf("%d%d",&n2,&m2);
		for(i=0;i<n2;++i)
		{
			for(j=0;j<m2;++j)
				scanf("%d",&b[i][j]); 
		}
		if(m==n2)
		{
			int k,temp;
			printf("YES\n");
			for(i=0;i<n;++i)
			{
				for(j=0;j<m2;++j)
				{
					temp=0;
					for(k=0;k<m;++k)
						temp+=a[i][k]*b[k][j];
					printf("%d",temp); 
					if(j==m2-1)
						printf("\n");
					else
						printf(" ");	
				}  
			} 
		} 
		else
			printf("NO\n"); 
	}
	return 0; 
} </span>