矩阵类 C++ 作为 matlab 运算符重载

Question

我有一个类定义一个维度为 mxn 的矩阵，如下所示：

class Matrix{
  protected:
  int m;
  int n;
  double* mat:
  public:
  // accessors, constructors, destructors, etc.
  void assignvalue(int, int, double);
}

现在，如果我需要在位置 i,j 上分配一个值，我有一个函数 assignvalue，它采用位置 i，j 并执行魔术并为该位置分配一个双精度值。但是，如果我能像您在 matlab 或 R 中那样分配一个值，那就太好了。

mymatrix(i,j) = 1.0;

你能告诉我我需要重载哪些运算符吗？谢谢。

Answer 1

假设您的 m 代表矩阵的高度，n 代表宽度，这样重载 operator() 应该可以解决问题：

double& Matrix::operator()(size_t i, size_t j)
{
    return mat[i*m+j];
}

const double& Matrix::operator()(size_t i, size_t j) const
{
    return mat[i*m+j];
}

这样，你可以这样写：

void f(Matrix & mymatrix ) {
   mymatrix(2, 3) = 5.0; // Calls the first function
   // ...
}
void f(Matrix const & m) {
   double a = m(1, 5); // Calls the second one
   //...
}