c++ - 使用带下标运算符的赋值运算符为 std::map 赋值

Question

我有一个矩阵类

template <typename T>
class Matrix
{
public:
  const size_t rows;
  const size_t cols;
  const std::map<std::array<int, 2>, T> data;

  Matrix(int a, int b) : rows(a), cols(b)
  {
  }
};

初始化如下:

Matrix<double> M(5,5);

创建一个 5x5 矩阵。

我想像这样给 map 赋值:

M[{1,2}] = 1;

我将如何以最具可读性的方式来做到这一点？我不确定如何让下标和赋值运算符协同工作。

Answer 1

让我们在 Matrix 中添加一些辅助别名

template <typename T>
class Matrix
{   
    // rather than stoping people changing the members via const
    // make them private
    size_t rows;
    size_t cols;
    map_type data;
public:
    using key_type = std::array<size_t, 2>;
    using mapped_type = T;
    using map_type = std::map<key_type, mapped_type>;

    Matrix(size_t r, size_t c) : rows(r), cols(c) { }

    const T& operator [](key_type key) const { return data[key]; }
          T& operator [](key_type key)       { return data[key]; }

    // other operations as appropriate
};