使用迭代器从双 for 循环构建向量

Question

我正在学习如何使用迭代器，但在概念化如何使用它们从双循环构造向量时遇到了麻烦。

这是一个示例：假设我想编写一个函数，该函数输入两个向量，并输出一个向量，该向量包含通过组合每个向量中的单个元素生成的所有可能的和（这是一个人为的示例）。这是一个工作版本：

#include <RcppArmadillo.h>
using namespace Rcpp;
// [[Rcpp::depends(RcppArmadillo)]]
// [[Rcpp::export]]
arma::colvec test(arma::colvec x, arma::colvec y) {
  arma::colvec out(x.size()*y.size());
  arma::colvec::iterator ix,iy;
  int count=0;
  for (ix=x.begin();ix!=x.end();ix++) {
    for (iy=y.begin();iy!=y.end();iy++) {
      out(count)=*ix+*iy;
      count+=1;
    }
  }
  return out;
}

/*** R
test(c(1:15),c(15:1))
*/

这很好用，但我觉得使用 count 作为 out 向量的临时迭代器似乎不太优雅，我觉得必须有更好的解决方案。我希望这样的事情会起作用：

arma::colvec::iterator ix,iy,io;
for (ix=x.begin(),io=out.begin();ix!=x.end(),io!=out.end();ix++,io++) {
  for (iy=y.begin();iy!=y.end();iy++,io++) {
    *io=*ix+*iy;
  }
}

这会使我的计算机崩溃，所以我猜我不能让io 对每个循环进行迭代。如果有人能指出在这种情况下什么是好的做法，我将不胜感激。

问候

Answer 1

每次嵌套循环完成时，您都会连续执行两次io++。这会使您的迭代器移动得比应有的更快，最终超出范围。 但是ix 也超出了范围。

从外部循环的迭代表达式中删除io++。那么它就相当于：

*(io++) = *ix + *iy;

io 没有其他增量。

现在，注意循环条件中的逗号运算符（这是导致崩溃的原因）：

ix != x.end(), io != out.end()

- 评估ix != x.end()，丢弃它的值，使用io != out.end() 的值。应该有逻辑 AND (&&)。如果，您应该在嵌套循环的条件下检查io != out.end()（这就是io 递增的地方）。

但是既然你这样构造out：

arma::colvec out(x.size() * y.size());

没有检查是安全的。

代码（C++11）：

auto io = out.begin();

for (auto x_elem : x)
  for (auto y_elem : y)
    *(io++) = x_elem + y_elem;