矩阵之转置、加、减、乘
结果:
打印原始数组:
[
[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;
}