c++ - 水平连接 std::vectors 的正确方法是什么？

Question

我有一个二维矩阵，其列大小在每次迭代后都会增加，但行大小保持不变。对于每次迭代，一个函数返回一个我想水平堆叠的矩阵。以下是我尝试过的代码，但我认为我在 big_mat[ii].insert 部分做错了。但我环顾四周，找到了垂直堆叠的代码，您可以在其中从 big_mat.end() 开始并执行插入。但我希望行大小保持固定。你能帮我解决这个问题吗？此外，最大最终大小将是 1,000,000 乘以 5,000 的数量级。

std::vector<std::vector<float> > big_mat;
big_mat.reserve(fixed_row_dim);

std::vector<std::vector<float> > small_mat;

for (some condition){
    // small_mat is always fixed_row_dim x some_dim
    small_mat = GetMat(params,fixed_row_dim);   
    for (int ii = 0; ii < fixed_row_dim; ii++){
        big_mat[ii].insert(big_mat[ii].end(),small_mat[ii].begin(),small_mat[ii].end());
    }
}

Answer 1

我已经尝试编译你的代码，当然有必要的版本，它工作得很好。那么问题是什么？您遇到了什么样的错误？

int main() {
  std::vector<std::vector<float> > big_mat;
  big_mat.reserve(5);

  std::vector<std::vector<float> > small_mat;

  for (int i = 0; i < 10; i++){
    // small_mat is always fixed_row_dim x some_dim
    std::vector<float> example = { 1, 2, 3, 4, 5 };
    std::vector<std::vector<float>> small_mat;

    for (int ii = 0; ii < 5; ii++) {
        small_mat.push_back(example);
    }
    for (int ii = 0; ii < 5; ii++){
        big_mat[ii].insert(big_mat[ii].end(), small_mat[ii].begin(), small_mat[ii].end());
    }
  }
}

但是我在 visual studio 2013 中尝试运行该代码时出现“调试断言失败”错误。原因是 big_mat 为空 - 大小小于迭代次数。更改 big_mat 的初始化:

 big_mat.reserve(5);

到:

  for (int ii = 0; ii < 5; ii++) {
    big_mat.push_back(*(new std::vector<float>()));
  }

解决了这个问题。我认为这是您的代码的问题。我将尝试获取有关 std::vector::reserve 函数的一些信息，并在接下来的几分钟内向您解释。

编辑: 正如@Zereges 指出的那样。建议的二维 vector 初始化方式会导致内存泄漏，安全且更性感的实现方式是:

std::vector<std::vector<float> > big_mat { 5, std::vector<float>{0f} }

说明:正如您在 documentation 中所读到的那样, reserve 函数

Requests that the vector capacity be at least enough to contain n elements.

只是保证给定大小的 vector 将适合内存而不需要重新分配。它不初始化任何东西，所以你的 big_mat vector vector 是空的，你试图在你的循环中迭代它。