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 <bits/stdc++.h> using namespace std; int a[101][101], b[101][101]; int main() { int ra, ca, rb, cb; scanf("%d%d", &ra, &ca); for(int i=0; i<ra; i++) { for(int j=0; j<ca; j++) { scanf("%d", &a[i][j]); } } scanf("%d%d", &rb, &cb); for(int i=0; i<rb; i++) { for(int j=0; j<cb; j++) { scanf("%d", &b[i][j]); } } if(ca == rb) { printf("%d %d\n", ra, cb); for(int i=0; i<ra; i++) { for(int j=0; j<cb; j++) { int sum = 0; for(int k=0; k<ca; k++) { sum += a[i][k]*b[k][j]; } if(j < cb-1) printf("%d ", sum); else printf("%d\n", sum); } } } else printf("Error: %d != %d\n", ca, rb); return 0; }