蓝桥杯 算法提高 ADV-208 矩阵相乘 线性代数

算法提高 矩阵相乘
时间限制：1.0s 内存限制：256.0MB
问题描述
　　小明最近在为线性代数而头疼，线性代数确实很抽象（也很无聊），可惜他的老师正在讲这矩阵乘法这一段内容。
　　当然，小明上课打瞌睡也没问题，但线性代数的习题可是很可怕的。
　　小明希望你来帮他完成这个任务。
　　现在给你一个ai行aj列的矩阵和一个bi行bj列的矩阵，
　　要你求出他们相乘的积（当然也是矩阵）。
　　(输入数据保证aj=bi,不需要判断)
输入格式
　　输入文件共有ai+bi+2行，并且输入的所有数为整数（long long范围内）。
　　第1行：ai 和 aj
　　第2~ai+2行：矩阵a的所有元素
　　第ai+3行：bi 和 bj
　　第ai+3~ai+bi+3行：矩阵b的所有元素
输出格式
　　输出矩阵a和矩阵b的积（矩阵c)
　　（ai行bj列）
样例输入
2 2
12 23
45 56
2 2
78 89
45 56
样例输出
1971 2356
6030 7141

分析：本题属于常规的矩阵相乘，输入ai行aj列的矩阵a与bi行bj列的矩阵b相乘，输出的矩阵c为ai行bj列。每一个元素的值为$c[i][j]+=a[i][k]\ast b[k][j];$，代码如下：

#include <iostream>
using namespace std;

int main()
{
	int ai, aj, bi, bj;
	int **a, **b, **c;
	
	//输入矩阵a
	cin >> ai >> aj;
	a = new int*[ai];
	for(int i = 0; i < ai; i++)
	{
		a[i] = new int[aj];
		for(int j = 0; j < aj; j++)
		{
			cin >> a[i][j];
		}
	}
	//输入矩阵b 
	cin >> bi >> bj;
	b = new int*[bi];
	for(int i = 0; i < bi; i++)
	{
		b[i] = new int[bj];
		for(int j = 0; j < bj; j++)
		{
			cin >> b[i][j];
		}
	}
	
	c = new int*[ai];
	for(int i = 0; i < ai; i++)
	{
		c[i] = new int[bj];
		for(int j = 0; j < bj; j++)
		{
			c[i][j] = 0;
			for(int k = 0; k < aj; k++)
			{
				c[i][j] += a[i][k] * b[k][j];
			}
			cout << c[i][j] << " ";
		}
		cout << endl;
	}
	
	return 0;
}