矩阵乘法

矩阵乘法

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

Submit  Status  Practice  HPU 1035

Description

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

Input

第一行输入一个整数 T T，代表有 T T组测试数据。

每组数据第一行输入两个整数 N,M N,M，代表矩阵 A A的行、列。

接下来 N N行，每行输入 M M个整数 a[][] a[][]。

之后一行输入两个整数 n,m n,m，代表矩阵 B B的行、列。

接下来 n n行，每行输入 m m个整数 b[][] b[][]。

注： 1<=T<=500，1<=N,m<=100，1<=M,n<=10，1<= 1<=T<=500，1<=N,m<=100，1<=M,n<=10，1<=矩阵元素 <=20 <=20。

Output

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

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

反之输出 NO 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

要掌握其写法

#include <iostream>
#include <cstdio>
int A[220][220],B[220][220],C[220][220];
int main()
{
	 int t;
	 scanf("%d",&t);
	 while(t--)
	 {
	int n,s,m,w;
	scanf("%d %d",&n,&s);
	for(int i = 0; i < n; i++)
	   for(int j = 0;  j < s ; j++)
	       scanf("%d",&A[i][j]);
	       scanf("%d %d",&w , &m);
	       for(int p = 0; p < w; p++)
	            for(int q = 0; q < m; q++)
	                   scanf("%d",&B[p][q]);
	      	if(s == w)
	    printf("YES\n");
	 else
	 {
	 	 printf("NO\n");
	 	 continue;	
	 }
	               int i  , j ,k ;
	     for(i = 0; i < n; i++)
		  {
               for(j = 0; j < m; j++) 
			 {
                  C[i][j] = 0;
            }
}
           for(i = 0; i < n; i++)
		    {
                for(k = 0; k < s; k++)
		  {
                if(A[i][k] == 0) 
			       continue;
               for(j = 0; j < m; j++) 
	    {
               C[i][j] = (C[i][j] + A[i][k] * B[k][j]);
        }
    }
}

				   for(int i = 0; i < n; i++)
				    for(int j = 0; j < m; j++)
				     {  
				          if( j == m-1)
				          printf("%d\n",C[i][j]);
				          else
				           printf("%d ",C[i][j]);
		
						}
				}
	       return 0;
}