矩阵乘法(两个矩阵相乘)
2017-08-14 16:19
155 查看
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
Sample Output
要搞清楚整个矩阵相乘的过程才能写好循环。
#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;
}
给定两个矩阵$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;
}
相关文章推荐
- 基于十字链表的两个稀疏矩阵相乘
- 简单的Tensoflow(1):创建两个向量并进行矩阵相乘
- 将两个矩阵相乘,A为x行y列的矩阵,B为y行z列的矩阵,A*B
- TF:Tensorflow之session会话的使用,定义两个矩阵,两种方法输出2个矩阵相乘的结果—Jason niu
- 是想两个矩阵相乘(并实现文件的输入输出)
- C++两个矩阵相乘
- 两个矩阵相乘问题
- 【面试题】两个n*n的矩阵相乘--采用一位数组表示
- 两个矩阵相乘但是结果不对,只有第一行的对,求解!
- 将两个矩阵相乘,A为x行y列的矩阵,B为y行z列的矩阵,A*B
- 两个矩阵相乘
- 第15周-两个二阶二维矩阵相乘
- Introduction to Java Programming编程题7.6<两个矩阵相乘>
- 使用指针数组实现这两个矩阵的相乘
- 求矩阵的转置矩阵和两个矩阵相乘
- 《两个二维数组(矩阵)相乘》
- [置顶] 解决两个矩阵相乘的好方法
- 两个矩阵相乘!
- 使用指针数组实现这两个矩阵的相乘
- 两个数组各个数相加或相乘变成一个矩阵求第K大