矩阵乘法（两个矩阵相乘）

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<cstring>  
#define MAX 105   
int a[MAX][11],b[11][MAX],c[MAX][MAX];  //定义三个数组 
int m,N,M,n,T;  
int main()  
{  
    int i,j;  
    while(scanf("%d",&T)!=EOF)  
    {  
        while(T--)  
        {  
            scanf("%d%d",&N,&M);  // N行M列 
            for(i=0;i<N;i++)  
                for(j=0;j<M;j++)  
                    scanf("%d",&a[i][j]);  //输入数组a
            scanf("%d%d",&n,&m);  //n行m列 
            for(i=0;i<n;i++)  
                for(j=0;j<m;j++)  
                    scanf("%d",&b[i][j]);  //输入数组b               
            if(M!=n)    //判断两个矩阵是否可以相乘 
			printf("NO\n");  
            else 
			{  
	            printf("YES\n");  
	            memset(c,0,sizeof(c));  //清空数组c 
	            int k;  
	            for(k=0;k<N;k++)  
	            {  
	                for(j=0;j<m;j++)  
	                {  
	                    for(i=0;i<n;i++)  
	                    {  
	                        c[k][j]+=a[k][i]*b[i][j];     //根据循环求数组c                 
	                    }  	                          
	                }  
            }         
            for(i=0;i<N;i++)  
                for(j=0;j<m;j++)  
                    printf("%d%c",c[i][j],j==m-1?'\n':' ');  //判断是否需要换行 
            }         
        }  
    }  
    return 0;  
}