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 2NO
<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>