Question

假设我有这个嵌套循环

for (int a=1; a<MAX_A; ++a)
  for (int b=1; b<MAX_B; ++b)
    for (int c=1; c<MAX_C; ++c)
    {
       do_something(a, b ,c);
    }

我在代码的各个部分重用了这个循环，改变了函数do_something。每次前三行重写都很无聊。例如，在python中，我创建了一个生成器，用于返回迭代器(1, 1, 1), (1, 1, 2), ...或类似itertools.product的内容。

在c ++中，我唯一想到的解决方案是定义一个宏。更好的东西？e

Answer 1

使用模板：

template<typename Func>
inline void do_something_loop(Func f)
{
    for (int a=1; a<MAX_A; ++a) 
      for (int b=1; b<MAX_B; ++b) 
        for (int c=1; c<MAX_C; ++c) 
        { 
           f(a, b ,c); 
        } 
}

可以使用与签名匹配的任何函数指针或函数对象调用它，例如：

void do_something(int a, int b, int c) { /* stuff */ }

do_something_loop(do_something);

或者使用函数对象：

struct do_something
{
    void operator()(int a, int b, int c) { /* stuff */ }
};

do_something_loop(do_something());

或者如果您的编译器支持C ++ 11，即使使用lambda：

do_something_loop([](int a, int b, int c) { /* stuff */ });

请注意，您也可以将f参数声明为带有签名void(*f)(int,int,int)的函数指针，而不是使用模板，但这不太灵活（它不适用于函数对象（包括std :: bind）或lambdas的结果。

Answer 2

制作一个辅助循环函数：

template<typename Work>
void loop(Work work) {
  for (int a=1; a<MAX_A; ++a)
    for (int b=1; b<MAX_B; ++b)
      for (int c=1; c<MAX_C; ++c)
      {
        work(a, b ,c);
      }
}

定义要完成的工作：

void some_work(int a, int b, int c) {
}

然后使用：

loop(some_work);

请注意，现在您不需要仅使用函数，但可以使用函数对象（operator()重载的对象）。

重复使用嵌套循环，无需复制和粘贴

2 个答案: