L1-048. 矩阵A乘以B
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
给定两个矩阵A和B,要求你计算它们的乘积矩阵AB。需要注意的是,只有规模匹配的矩阵才可以相乘。即若A有Ra行、Ca列,B有Rb行、Cb列,则只有Ca与Rb相等时,两个矩阵才能相乘。
输入格式:
输入先后给出两个矩阵A和B。对于每个矩阵,首先在一行中给出其行数R和列数C,随后R行,每行给出C个整数,以1个空格分隔,且行首尾没有多余的空格。输入保证两个矩阵的R和C都是正数,并且所有整数的绝对值不超过100。
输出格式:
若输入的两个矩阵的规模是匹配的,则按照输入的格式输出乘积矩阵AB,否则输出“Error: Ca != Rb”,其中Ca是A的列数,Rb是B的行数。
输入样例1:2 3 1 2 3 4 5 6 3 4 7 8 9 0 -1 -2 -3 -4 5 6 7 8输出样例1:
2 4 20 22 24 16 53 58 63 28输入样例2:
3 2 38 26 43 -5 0 17 3 2 -11 57 99 68 81 72输出样例2:
Error: 2 != 3
#include<iostream>
#include<vector>
using namespace std;
typedef vector<int> vi;
void input(vector<vi> &s,int n,int m){
int t;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++){
scanf("%d",&t);
s[i].push_back(t);
}
}
int main(){
int n1,m1,n2,m2;
scanf("%d%d",&n1,&m1);
vector<vi> v(n1);
input(v,n1,m1);
scanf("%d%d",&n2,&m2);
vector<vi> s(n2);
input(s,n2,m2);
vector<vi> ans(n1);
if(m1!=n2)
printf("Error: %d != %d\n",m1,n2);
else{
int t;
printf("%d %d\n",n1,m2);
for(int i=0;i<n1;i++){
for(int k=0;k<m2;k++){
t=0;
for(int j=0;j<m1;j++){
t+=v[i][j]*s[j][k];
}
ans[i].push_back(t);
}
}
for(int i=0;i<n1;i++){
int j;
for(j=0;j<m2-1;j++)
printf("%d ",ans[i][j]);
printf("%d\n",ans[i][j]);
}
}
}