Как перегрузить оператор индекса массива для класса обертки 2D массива?
#define ROW 3
#define COL 4
class Matrix
{
private:
int mat[ROW][COL];
//.....
//.....
};
int main()
{
Matrix m;
int a = m[0][1]; // reading
m[0][2] = m[1][1]; // writing
}
Я думаю непосредственно это не возможный перегрузиться [] [].
Я думаю, что должен сделать это косвенно, но как реализовать его?
2 ответа
Проще всего использовать operator(), так как он допускает несколько параметров.
class M
{
public:
int& operator()(int x,int y) {return at(x,y);}
// .. Stuff to hold data and implement at()
};
M a;
a(1,2) = 4;
Проще всего, что первый operator[] возвращает промежуточный объект, а второй operator[] возвращает значение из массива.
class M
{
public:
class R
{
private:
friend class M; // Only M can create these objects.
R(M& parent,int row): m_parent(parent),m_row(row) {}
public:
int& operator[](int col) {return m_parent.at(m_row,col);}
private:
M& m_parent;
int m_row;
};
R operator[](int row) {return R(*this,row);}
// .. Stuff to hold data and implement at()
};
M b;
b[1][2] = 3; // This is shorthand for:
R row = b[1];
int& val = row[2];
val = 3;
Так как вы хотите хранить свои элементы в массивах фиксированного размера, то это будет довольно просто:
#define ROWS 3
#define COLS 4
typedef int row_type[COLS];
class matrix {
row_type elements[ROWS];
public:
...
row_type const& operator[](int r) const {return elements[r];}
row_type & operator[](int r) {return elements[r];}
...
};
Это должно сработать.
Кроме того, вы можете захотеть заменить #defines на соответствующие константы или использовать параметры шаблона для типа (int) и размера (3x4), чтобы сделать ваш класс матрицы более универсальным. Если вы хотите поддерживать динамические размеры, ваш operator[] должен возвращать прокси-объекты. Это возможно, но вы, вероятно, должны предпочесть operator() с двумя параметрами индекса для доступа к элементам.