我正在慢慢学习boost
,我正在尝试找到一种简单的方法来转换以下C ++代码段:
for(int i=-n;i<n+1;i++) {
for(int j=-n;j<n+1;j++) {
for(int k=-n;k<n+1;k++) {
cout << i << ' ' << j << ' ' << k << endl;
}
}
}
进入我可以遍历的单个迭代器。用我的母语python
(我可以称之为吗?),这是一个使用itertools
的单行语言:
itrtools.product(range(-n,n+1),repeat=3)
完整的答案将提供最小的工作示例和文档链接,以便我可以RTFM。
答案 0 :(得分:3)
BOOST_PP_SEQ_FOR_EACH_PRODUCT
中Boost.Preprocessor可以在预处理步骤中执行此操作。
BOOST_PP_SEQ_FOR_EACH_PRODUCT宏为几个seqs的每个笛卡尔积重复一个宏。
但我想这不是你想要的。
如果一个可重用的小代码没问题,那么您可以使用Function Input Iterator中的Boost.Iterator生成给定范围的笛卡尔积。
<强>发电机强>
class product_generator
{
public:
typedef std::vector<int> result_type;
product_generator (int lower, int upper, unsigned int repeat)
: m_lower(lower), m_upper(upper)
{
for(unsigned int i = 0; i < repeat; ++i)
{
m_iters.push_back(m_lower);
}
};
std::vector<int> operator() ()
{
for(int& i : m_iters)
{
if(++i >= m_upper)
i = m_lower;
else
break;
}
std::vector<int> res;
for(int i : m_iters)
res.push_back(i);
return res;
};
private:
int m_lower;
int m_upper;
std::vector<int> m_iters;
};
使用此生成器,您可以执行以下操作:
product_generator p(lower, upper, repeat);
auto bgn = boost::make_function_input_iterator(p, (double)0);
bgn
是单个迭代器,您可以循环生成由输入范围形成的输入序列的笛卡尔积。
一个完整的工作示例:
#include <vector>
#include <iostream>
#include <boost/iterator/function_input_iterator.hpp>
#include <math.h>
int main ()
{
int lower = 1;
int upper = 4;
unsigned int repeat = 3;
product_generator p(lower, upper, repeat);
for(
auto bgn = boost::make_function_input_iterator(p, (double)0);
bgn != boost::make_function_input_iterator(p, pow(upper-lower, repeat));
++bgn
)
{
for(int i : *bgn)
{
std::cout << i << " ";
}
std::cout << std::endl;
}
}