B

点击打开链接

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

22 21 11 12 31 1 11 1 12 31 1 11 1 12 21 11 1 

Sample Output

YES2 2 22 2 2NO 

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;int a[105][105],b[105][105],c[105][105];int main(){int t;scanf("%d",&t);while(t--){int N,M,n,m,i,j;scanf("%d %d",&N,&M);for(i=1;i<=N;i++)  for(j=1;j<=M;j++)    scanf("%d",&a[i][j]);scanf("%d %d",&n,&m);for(i=1;i<=n;i++)  for(j=1;j<=m;j++)    scanf("%d",&b[i][j]);if(M!=n)  printf("NO\n");else{  memset(c,0,sizeof(c));  int l,k;  for(i=1;i<=N;i++)    for(j=1;j<=m;j++){  l=k=0;  while(l<M&&k<n)  {  l++;  k++;  c[i][j]+=a[i][l]*b[k][j];  }}  printf("YES\n");  for(i=1;i<=N;i++)    for(j=1;j<=m;j++)  printf("%d%c",c[i][j],j==m?'\n':' ');}   }}