C ++ Armadillo:使用迭代器进行双循环

时间:2016-01-06 22:41:31

标签: c++ armadillo

在双重for循环中使用迭代器的最佳方法是什么。对于单循环,显而易见的方式似乎是:

arma::vec test = arma::ones(10);
for(arma::vec::iterator i = test.begin(); i != test.end(); ++i){
   int one = *i;
}

所以我想改变以下内容:

arma::mat test = arma::ones(10,10);
for (int i = 0; i < test.n_rows; i++){
 for (int j = 0; j < test.n_cols; j++){
  int one = test(i,j);
 }
}

使用迭代器而不是整数索引。谢谢你的任何建议。

2 个答案:

答案 0 :(得分:5)

建议在访问矩阵中的元素时仍然使用单个循环。 Armadillo以column-major格式存储数据(为了与LAPACK兼容),因此迭代器将沿着矩阵中的每一列传播。

arma::mat A(4, 5, arma::fill::randu);
A.print("A:");

// C++98 

arma::mat::iterator it_end = A.end();

for(arma::mat::iterator it = A.begin(); it != it_end; ++it)
   {
   std::cout << (*it) << std::endl;
   }

// C++11

for(const auto& val : A)
   {
   std::cout << val << std::endl;
   }

如果您真的想要使用双循环,请使用.begin_col().end_col()

// C++11

for(arma::uword c=0; c < A.n_cols; ++c)
  {
  auto it_end = A.end_col(c);

  for(auto it = A.begin_col(c); it != it_end; ++it)
    {
    std::cout << (*it) << std::endl;
    }
  }

最后,.for_each()函数是使用迭代器的替代方法:

 // C++11

 A.for_each( [](arma::mat::elem_type& val) { std::cout << val << std::endl; } );

答案 1 :(得分:1)

有三件事情浮现在脑海中,而不仅仅是双重迭代,而是任何类型的迭代(只需将下面的所有内容应用两次......)

1)如果容器的内容在循环期间没有改变,则调用end()一次并保存它,并在循环的每次迭代中将迭代器与保存的值进行比较。

不太好:

for(arma::vec::iterator i = test.begin(); i != test.end(); ++i)

如上所述,end()在每个循环的迭代中被技术调用,并且它的返回值与循环迭代器的当前值进行比较。

2)

更好:

const arma::vec::iterator e=test.end();

for(arma::vec::iterator i = test.begin(); i != e; ++i)

这有更好的机会进行优化,特别是使用const限定符(尽管编译器可能能够自己解决它,如果e永远不会被修改,无论如何)。

3)

最佳:

for(const auto &value:test)

C ++ 11迭代语法告诉编译器究竟发生了什么,为编译器提供了最好的机会来使用它编程知道的所有容器迭代技巧。