OJ-矩阵乘法

矩阵乘法
Description

矩阵乘法是线性代数中最基本的运算之一。

给定三个矩阵A，B，C，请编写程序判断A×B=C是否成立。

InputC
输入包含多组数据。每组数据的格式如下：

第一行包括两个整数p和q，表示矩阵A的大小。后继p行，每行有q个整数，表示矩阵A的元素内容。

紧接着用相同的格式给出矩阵B和矩阵C。

输入数据的最后一行是两个0，你的程序处理到这里时就应该退出了。

输入数据中所有的整数绝对值都不超过100。

Output
对每一组输入数据，你的程序都要输出单独一行字符。

如果A×B=C成立则输出”Yes”

如果A和B根本不能相乘，或A×B=C不成立，则输出”No”。注意大小写。

思路

假设 A矩阵为a1×a2，B矩阵为b1×b2，C矩阵为c1×c2
首先判断能否相乘，即a2是否等于b2，如果不等于，便输出"No"
再判断a1是否等于c1，b2是否等于c2
如果不满足条件，则A×B=C必不成立。输出"No"
在上述条件都满足下
则需要比较A×B是否等于C，即对应位置的元素是否相等
只要有一个不相等，则输出"No"
现在只剩下一步——矩阵乘法
对于本题来说，就是拿A矩阵的行元素一一与B矩阵的列元素相乘的和
A的第一行与B的第一列，构成C[1][1]（对于数组，开头是0，即C[0][0]）
A的第一行与B的第二列，构成C[1][2]（对于数组，开头是0，即C[0][1]）
首先两个大循环
第一个大循环，用于找到矩阵A的行，第二个大循环，用于找到矩阵B的列
即

for（int i=0;i<a1;i++){
    for(int j=0;j<b2;j++){
    }
}

紧接着就是用一个变量 t，用于行与列元素一一相乘
假设是A的第一行，就是A[0][t]*B[t][j]
即在矩阵A中第i行的第t个元素应该与矩阵B中第t行的第j个元素相乘
因为只要一个元素不同，等式便不成立，所以本处用了一个布尔变量，一旦不同，终止循环，避免时间浪费

代码

#include <iostream>
using namespace std;

int main()
{
	int a[100][100];
	int b[100][100];
	int c[100][100];
	int a1, a2;
	while (cin >> a1 >> a2) {
		if (a1 == 0 && a2 == 0)
			break;
		for (int i = 0; i < a1; i++) {
			for (int j = 0; j < a2; j++)
				cin >> a[i][j];
		}
		int b1, b2;
		cin >> b1 >> b2;
		for (int i = 0; i < b1; i++) {
			for (int j = 0; j < b2; j++)
				cin >> b[i][j];
		}
		int c1, c2;
		cin >> c1 >> c2;
		for (int i = 0; i < c1; i++) {
			for (int j = 0; j < c2; j++)
				cin >> c[i][j];
		}
		if (a1 != c1 || b2 != c2||a2!=b1)
			cout << "No" << endl;
		else {
			int temp = 0;
			bool IsRight = true;
			for (int i = 0; i < a1; i++) {
				if (!IsRight)
					break;
				for (int j = 0; j < b2;j++) {
					if (!IsRight)
						break;
					for (int t = 0; t < a2; t++) {
						temp =temp+ a[i][t] * b[t][j];
					}
					if (c[i][j] != temp)
						IsRight = false;
					temp = 0;
				}
			}
			if (IsRight)
				cout << "Yes" << endl;
			else
				cout << "No" << endl;
		}
	}
}