Оператор [] в двумерном векторе

Question

Я хочу создать оператор [] в случае двумерного вектора. После поиска я обнаружил, что невозможно передать два аргумента. Как я могу получить значение m_matrix[i][j] в основном?

соответствующий код:

class MyMatrix 
{ 
    public:   // Methods 

     MyMatrix(); 
     ~MyMatrix(); 

     int operator[] (int n); 

    private:   // Attributes 

     int m_n; 
     int m_m; 

     std:: vector <std:: vector <int> > m_matrix; 

}; 

int MyMatrix::operator[](int n, int m) // in the cpp 
{ 
    if (n>=0 && m>=0 && n<=m_n && m<=m_m) 
    { 
     return m_matrix[n-1][m-1]; 
    } 
    else 
    { cout<<"******************"<<endl; 
     cout<<"No valid index"<<endl; 
     cout<<"******************"<<endl; 
     return 0; 
    } 
} 

... 

mat_test1[2][2]; // for example in the main 

Что случилось с этим?

Answer 1

Чтобы возобновить комментарий, вы можете сделать:

class MyMatrix 
{ 
public: 

    // Other methods 

    const std::vector<int>& operator[] (int m) const { return m_matrix.at(m); } 
    std::vector<int>& operator[] (int m) { return m_matrix.at(m); } 

    int operator() (int m, int n) const { return m_matrix.at(m).at(n); } 
    int& operator() (int m, int n) { return m_matrix.at(m).at(n); } 

private: 
    std::vector<std::vector<int>> m_matrix; 
}; 

Примечания: Я использовал at вместо [] использовать чек от вектора и поэтому он бросает исключение для отказа от связанного доступа.

И затем использовать его:

MyMatrix matrix(5, 4); // size of the matrix from constructor 

matrix[0][1] = 42; // MyMatrix::operator [] followed by std::vector<int>::operator[] 
matrix(2, 2) = 42; // MyMatrix::operator() (int, int); 

Live example.

Answer 2

Существует несколько способов, но в конце концов все они сводятся к использованию прокси-сервера .

По разным причинам обычная реализация матрицы не std::vector<std::vector<T>>, а просто std::vector<T>, с соответствующего расчета индекса. В этом случае:

class Matrix 
{ 
    int myRows; 
    int myColumns; 
    std::vector<int> myData; 

    class Proxy 
    { 
     Matrix* myOwner; 
     int myRow; 
    public: 
     Proxy(Matrix* owner, int row) 
      : myOwner(owner) 
      , myRow(row) 
     { 
     } 
     int& operator[](int column) 
     { 
      return myOwner->get(myRow, column); 
     } 
    }; 
public: 
    Matrix(int rows, int columns) 
     : myRows(rows) 
     , myColumns(columns) 
     , myData(rows * columns) 
    { 
    } 

    int& get(int row, int column) 
    { 
     assert(0 <= row && row < myRows 
       && 0 <= column && column < myColumns); 
     return myData[ row * myColumns + column ]; 
    } 

    Proxy operator[](int row) { return Proxy(this, row); } 
}; 

В самом деле, вы хотите еще несколько перегруженных (и, возможно, два прокси) на ручки сопзЬ и неконстантные перегрузки. Но это общий шаблон для таких вещей.

Имейте в виду, что существуют два соглашения относительно таких перегрузок. Одним из них является сказать, что [] возвращает проекцию, и что правильное решения для обработки индексации в многомерные структуры является перегрузкой operator() (который может быть выполнен, чтобы принимать произвольное число параметров ). Я лично предпочитаю решение [][], но это предпочтение не универсально; много людей с более математическим фоном предпочитают использовать ().