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 2NO
解析:
注意数组的初始化,博主就因为这个WA了一下。还有就是矩阵可以相乘的条件就是
矩阵A的列数等于矩阵B的行数。
程序如下:
#include<cstdio> int main() { int t; scanf("%d",&t); while(t--) { int n,m,p,q; int a[103][103],b[103][103],c[103][103]={0}; scanf("%d%d",&m,&n); int i,j,k; for(i=0; i<m; i++) for(j=0; j<n; j++) scanf("%d",&a[i][j]); scanf("%d%d",&p,&q); for(i=0; i<p; i++) for(j=0; j<q; j++) scanf("%d",&b[i][j]); if(n==p) { printf("YES\n"); for(i=0; i<m; i++) for(j=0; j<n; j++) for(k=0; k<q; k++) c[i][k]+=a[i][j]*b[j][k]; for(i=0; i<m; i++) for(k=0; k<q; k++) printf("%d%c",c[i][k],k==q-1?'\n':' '); } else printf("NO\n"); } return 0; }