矩阵之转置、加、减、乘

矩阵之转置、加、减、乘

结果：

打印原始数组：
[
[1 2 3 ]
[4 5 6 ]
[7 8 9 ]
[3 4 5 ]
]
转置后：
[
[1 4 7 3 ]
[2 5 8 4 ]
[3 6 9 5 ]
]

d:
[
[1 2 3 ]
[4 5 6 ]
]
e:
[
[1 2 3 ]
[4 5 6 ]
]
d e 相加：
[
[2 4 6 ]
[8 10 12 ]
]
d e 相减：
[
[0 0 0 ]
[0 0 0 ]
]

g:
[
[1 2 ]
[4 5 ]
]
h:
[
[1 2 3 ]
[4 0 6 ]
]
g h 相乘：
[
[9 2 15 ]
[24 8 42 ]
]

#include<iostream>
#include<vector>

// 打印
void display(std::vector < std::vector<int>>& V) {
	std::cout << "[" << '\n';
	for (auto& v : V) {
		std::cout << "[";
		for (auto& i : v) {
			std::cout << i << ' ';
		}
		std::cout << "]" << '\n';
	}
	std::cout << "]" << '\n';
}

// 转置
std::vector < std::vector<int>> traMatrix(std::vector < std::vector<int>>& V) {
	std::vector < std::vector<int>> v(V[0].size(), std::vector<int>(V.size(), 0)); // 初始化二维数组大小
	for (size_t i = 0; i < V.size(); ++i) {
		for (size_t j = 0; j < V[0].size(); ++j) {
			v[j][i] = V[i][j];
		}
	}
	return v;
}

// 相加 c = a + b
std::vector < std::vector<int>> addMatrix(std::vector < std::vector<int>>& a, std::vector < std::vector<int>>& b) {
	std::vector < std::vector<int>> c(a.size(), std::vector<int>(a[0].size(), 0));
	for (size_t i = 0; i < a.size(); ++i) {
		for (size_t j = 0; j < a[0].size(); ++j) {
			c[i][j] = a[i][j] + b[i][j];
		}
	}
	return c;
}

// 相减 c = a - b
std::vector < std::vector<int>> subMatrix(std::vector < std::vector<int>>& a, std::vector < std::vector<int>>& b) {
	std::vector < std::vector<int>> c(a.size(), std::vector<int>(a[0].size(), 0));
	for (size_t i = 0; i < a.size(); ++i) {
		for (size_t j = 0; j < a[0].size(); ++j) {
			c[i][j] = a[i][j] - b[i][j];
		}
	}
	return c;
}

// 相乘 c = ab
std::vector < std::vector<int>> mulMatrix(std::vector < std::vector<int>>& a, std::vector < std::vector<int>>& b) {
	std::vector < std::vector<int>> c(a.size(), std::vector<int>(b[0].size(), 0));
	for (size_t i = 0; i < a.size(); ++i) {
		for (size_t j = 0; j < b[0].size(); ++j) {
			for (size_t k = 0; k < a[0].size(); ++k) {
				c[i][j] += a[i][k] * b[k][j];
			}
		}
	}
	return c;
}

// 主函数
int main() {
	std::vector < std::vector<int>> a{ {1,2,3}, {4,5,6}, {7,8,9} ,{3,4,5} };
	std::vector < std::vector<int>> d{ {1,2,3}, {4,5,6} };
	std::vector < std::vector<int>> e{ {1,2,3}, {4,5,6} };
	std::vector < std::vector<int>> g{ {1,2}, {4,5} };
	std::vector < std::vector<int>> h{ {1,2,3}, {4,0,6} };

	std::cout << "打印原始数组：" << '\n';
	display(a);

	std::cout << "转置后：" << '\n';
	auto b(traMatrix(a));
	display(b);

	std::cout << '\n' << "d:" << '\n';
	display(d);
	std::cout << "e:" << '\n';
	display(e);

	std::cout << "d e 相加：" << '\n';
	auto add(addMatrix(d, e));
	display(add);

	std::cout << "d e 相减：" << '\n';
	auto sub(subMatrix(d, e));
	display(sub);

	std::cout << '\n' << "g:" << '\n';
	display(g);
	std::cout << "h:" << '\n';
	display(h);
	std::cout << "g h 相乘：" << '\n';
	auto mul(mulMatrix(g, h));
	display(mul);

	system("pause");
	return 0;
}