Вставка элементов в многомерный вектор

vector<vector<int>> sort_a;
vector<int> v2;
vector<int> v3;

for (int i=0; i<4; ++i) {
v2.push_back(i);

  for (int j=0; j<4; ++j) {
  v3.push_back(j);
  sort_a.push_back(v2);
  sort_a.push_back(v3);
  }

}

вектор sort_a должен быть массивом 4x4, вместо этого выход 31x1 с большим количеством пустых элементов, как вставить элементы в многомерный вектор ?

2 ответов


не думаю, что это многомерный вектор, думаю, это как вектор векторов.

int n = 4;
std::vector<std::vector<int>> vec(n, std::vector<int>(n));

// looping through outer vector vec
for (int i = 0; i < n; i++) {
  // looping through inner vector vec[i]
  for (int j = 0; j < n; j++) {
    (vec[i])[j] = i*n + j;
  }
}

я включил в скобки (vec[i])[j] просто для понимания.

Edit:

если вы хотите заполнить свой вектор через push_back, вы можете создать временный вектор во внутреннем цикле, заполнить его, а затем push_back его в свой вектор:

for (int i = 0; i < n; i++) {
  std::vector<int> temp_vec;

  for (int j = 0; j < n; j++) {
    temp_vec.push_back(j);
  }

  vec.push_back(temp_vec);
}

однако, push_back вызовы приводят к более медленному коду, так как не только вам нужно перераспределить ваш вектор все время, но также вы должны создать временный и скопируйте его.


a vector<vector<int>> не является лучшей реализацией для многомерного хранилища. Следующая имплантация работает для меня.

template<typename T>
class array_2d {
    std::size_t data;
    std::size_t col_max;
    std::size_t row_max;
    std::vector<T> a;
public:
    array_2d(std::size_t col, std::size_t row) 
         : data(col*row), col_max(col), row_max(row), a(data)
    {}

    T& operator()(std::size_t col, std::size_t row) {
        assert(col_max > col && row_max > row)
        return a[col_max*col + row];
    }
};

использовать случае:

array_2d<int> a(2,2);
a(0,0) = 1;
cout << a(0,0) << endl;

Это решение аналогично описанному здесь.