Question

我有这样的2D矢量：vector<vector<int>>。我正在使用迭代器按行遍历它但是按列遍历它的最佳做法是什么？

这是我用于按行迭代的代码：

vector<vector<int>> vMatrix (4, vector<int>(4));
vector<vector<int>>::iterator itRow;
vector<int>::iterator itCol;

for (itRow = vMatrix.begin(); itRow != vMatrix.end(); itRow++)
{
    for (itCol = itRow->begin(); itCol != itRow->end(); itCol++) 
    {
        // do some stuff
    }
}

亲切的问候， Milen Vichev

Answer 1

在这种情况下可能的解决方案：

for (int col = 0; col < 4; col++)
{
    for (int row = 0; row < 4; row++) 
    {
        // do some stuff
    }
}

Answer 2

我认为这样做的一种方法是通过矩阵的转置：

std::vector<std::vector<int>> transpose(const std::vector<std::vector<int>> &m)
{
  using std::vector;

  vector<vector<int>> result(m[0].size(), vector<int>(m.size()));

  for (vector<int>::size_type i(0); i < m[0].size(); ++i)
    for (vector<int>::size_type j(0); j < m.size(); ++j)
      result[i][j] = m[j][i];

  return result;
}

和

std::vector<std::vector<int>>::iterator itCol;
std::vector<int>::iterator itRow;
std::vector<std::vector<int>> t(transpose(vMatrix));

for (itRow = t.begin(); itRow != t.end(); itRow++)
  for (itCol = itRow->begin(); itCol != itRow->end(); itCol++)
  {
    // ...
  }

循环体内没有什么需要改变，但它很慢。

您可以获得一些速度修改transpose以返回＆＃34;视图＆＃34;矩阵的转置：

std::vector<std::vector<int *>> transpose(const std::vector<std::vector<int>> &)

无论如何，即使这个解决方案比通过operator[]访问元素要慢。

可能，如果代码重构是一个选项，一个好的解决方案是从向量向量更改为1D向量以获得更好的代码局部性（类似https://stackoverflow.com/a/15799557/3235496）。

C ++ 2D矢量按列搜索

2 个答案: