在 C++ 中,你可以使用 std::complex
类来表示复数
#include
#include
#include
using namespace std;
typedef complex Complex;
typedef vector> Matrix;
Matrix multiply(const Matrix &A, const Matrix &B) {
int rows_A = A.size();
int cols_A = A[0].size();
int rows_B = B.size();
int cols_B = B[0].size();
if (cols_A != rows_B) {
throw invalid_argument("矩阵 A 的列数必须等于矩阵 B 的行数");
}
Matrix result(rows_A, vector(cols_B, Complex(0, 0)));
for (int i = 0; i < rows_A; ++i) {
for (int j = 0; j < cols_B; ++j) {
for (int k = 0; k < cols_A; ++k) {
result[i][j] += A[i][k] * B[k][j];
}
}
}
return result;
}
int main() {
Matrix A = {
{Complex(1, 2), Complex(3, 4)},
{Complex(5, 6), Complex(7, 8)}
};
Matrix B = {
{Complex(9, 10), Complex(11, 12)},
{Complex(13, 14), Complex(15, 16)}
};
Matrix result = multiply(A, B);
for (const auto &row : result) {
for (const auto &elem : row) {
cout << elem << " ";
}
cout << endl;
}
return 0;
}
这个程序首先定义了一个 Matrix
类型,它是一个二维向量,其元素为 Complex
类型。然后,我们实现了一个名为 multiply
的函数,它接受两个矩阵作为参数并返回它们的乘积。最后,在 main
函数中,我们创建了两个矩阵并计算它们的乘积。
注意:这个程序没有进行任何错误检查,例如检查矩阵是否为空或具有相同的行数和列数。在实际应用中,你可能需要添加这些检查以确保程序的健壮性。