使用迭代器从双 for 循环构建向量
Building vector from a double for loop using iterators
我正在学习如何使用迭代器,但我无法概念化如何使用它们从双循环构造向量。
这是一个示例:假设我想编写一个函数,它输入两个向量,并输出一个向量,该向量包含通过组合每个向量中的单个元素生成的所有可能的总和(这是一个人为的示例)。这是一个工作版本:
#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
在每个循环中迭代。如果有人能指出在这种情况下什么是好的做法,我将不胜感激。
此致
每次嵌套循环完成时,您都在连续执行两次 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;
我正在学习如何使用迭代器,但我无法概念化如何使用它们从双循环构造向量。
这是一个示例:假设我想编写一个函数,它输入两个向量,并输出一个向量,该向量包含通过组合每个向量中的单个元素生成的所有可能的总和(这是一个人为的示例)。这是一个工作版本:
#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
在每个循环中迭代。如果有人能指出在这种情况下什么是好的做法,我将不胜感激。
此致
每次嵌套循环完成时,您都在连续执行两次 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;