L1-048 矩阵A乘以B (15分)(详解)

给定两个矩阵A和B,要求你计算它们的乘积矩阵AB。需要注意的是,只有规模匹配的矩阵才可以相乘。即若A有R​a​​ 行、C​a​​ 列,B有R​b​​ 行、C​b​​ 列,则只有C​a​​ 与R​b​​ 相等时,两个矩阵才能相乘。
输入格式:

输入先后给出两个矩阵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

想要做出此题,除了要熟练矩阵的乘法,还需要一定的分析能力,可以再草稿纸上列出两个矩阵进行模拟,如图在这里插入图片描述
此题最关键就是最后的三个循环,可以在纸上进行模拟,第一次没做出来没有关系,只要记住了以后就不会忘了。
此题不需要定义相乘完的数组cc,算出一个输出一个即可,否则会运行超时!

#include <iostream>
#include <string>
#include<algorithm>
#include<bits/stdc++.h>
#include<stack>
#include<set>
#include <vector>
#include <map>
using namespace std;
int main() {
    
    
	int xa,ya,xb,yb;
	cin>>xa>>ya;
	int a[xa][ya];
	for(int i=0;i<xa;i++){
    
    
		for(int j=0;j<ya;j++){
    
    
			cin>>a[i][j];
		}
	}
	cin>>xb>>yb;
	int b[xb][yb];
	for(int i=0;i<xb;i++){
    
    
		for(int j=0;j<yb;j++){
    
    
			cin>>b[i][j];
		}
	}
	if(ya==xb){
    
    
		cout<<xa<<" "<<yb<<endl;
		for(int i=0;i<xa;i++){
    
    
			for(int j=0;j<yb;j++){
    
    
				int sum=0;
				for(int k=0;k<ya;k++){
    
    //k代表的是a的列和b的行,进行相加求和操作
					sum+=a[i][k]*b[k][j];
				}
				if(j<yb-1){
    
    
					cout<<sum<<" ";
				}
				else{
    
    
					cout<<sum<<endl;
				}
			}
		}
	}
	else{
    
    
		cout<<"Error: "<<ya<<" != "<<xb;
	}
	
}

猜你喜欢

转载自blog.csdn.net/weixin_45962741/article/details/113378656