什么是水平连接std :: vectors的正确方法?

时间:2016-01-17 11:30:01

标签: c++ std stdvector

我有一个2D矩阵,其列大小在每次迭代后都会增加,但行大小保持不变。对于每次迭代,函数返回一个我希望水平堆叠的矩阵。以下是我尝试的代码,但我认为我在big_mat [ii] .insert部分做错了。但我环顾四周,找到了垂直堆叠的代码,你可以从big_mat.end()开始执行插入。但我希望行大小保持固定。你能帮我解决这个问题吗?最大最终尺寸也是5,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());
    }
}

1 个答案:

答案 0 :(得分:2)

我已经尝试编译你的代码,当然还有必要的版本,它运行得很好。那么问题是什么,你会得到什么样的错误?

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());
    }
  }
}

然而我得到了#34;调试断言失败&#34;尝试在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指出,

编辑。建议的二维向量初始化方式会导致内存泄漏,安全性和更性感的方法是:

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

说明,因为您可以阅读documentationreserve函数

  

请求向量容量至少足以包含n个元素。

只保证给定大小的向量适合内存而无需重新分配。它没有初始化任何内容,因此你的big_mat向量向量是空的,你试图在循环中迭代它。