Matrix Multiplication
题目描述
Johnny and John are good friends. Johnny is going to take the entrance exams for postgraduate schools.
Recently, he is reviewing Linear Algebra. Johnny always says that he is very skillful at matrix
multiplication. As we all know that, if we know matrix a (m rows and n columns) and b (n rows and t
columns), the result matrix c (m rows and t columns) can be calculated by expression
c (i, j) = a(i,k)*b(k,j),1<=k<=n.(1<= i <= m, 1 <= j <= t).
But John wants to make things difficult. He wants Johnny to calculate matrix c using expression
c (i, j) = a(i,k)*b(k,j) (1<= i <= m, 1 <= j <= t, i + k is odd and k + j is even).
We consider that 2 is odd and even. At the beginning, c (i, j) =0.
输入
The first line of input is the number T of test case.
The first line of each test case contains three integers, m, n, t, indicates the size of the matrix a and b. We
always assume that the rows of matrix a equals to matrix b's columns.
The next m lines describe the matrix a. Each line contains n integers.
Then n lines followed, each line contains t integers, describe the matrix b.
1<= T <= 10, 1 <= m, n, t <= 100, 0 <= a (i, j) <= 100, 0 <= b (i, j) <= 100.
输出
For each test case output the matrix c. The numbers are separated by spaces. There is no space at the end of
each line.
样例输入
1 2 2 2 1 2 2 3 2 3 3 0
样例输出
2 0 4 0
代码如下
#include<iostream>
using namespace std;int a[105][105];
int b[105][105];
int c[105][105];
int main()
{
int m,n,t,k,s,i,j;
cin>>k;//测试数据的个数
while(k--)
{
cin>>m>>n>>t;
for(i=1;i<=m;i++)
{
for(j=1;j<=n;j++)
cin>>a[i][j];//输入矩阵a;
}
for(i=1;i<=n;i++)
{
for(j=1;j<=t;j++)
cin>>b[i][j];//输入矩阵b;
}
for (i=1;i<=m;i++)
{
for (j=1;j<=t; j++)
{ c[i][j]=0;
for (s=1; s<=n; s++)
{
if (((i+s)%2==1||(i+s==2))&&((j+s)%2==0||(j+s==2)))
{
c[i][j]+=a[i][s]*b[s][j];
}
}
}
}
for(i=1;i<=m;i++)
{
for(int j=1;j<=t-1;j++)
{
cout<<c[i][j]<<" ";
}
cout<<c[i][t]<<endl;
}
}
return 0;
}