C++中的循环就像在python中一样(基于范围的)

Nik*_*ond 5 c++ python range cycle c++11

像python一样在c ++中进行循环的最简单方法是什么?

for i in range(10): #or range(4, 10, 2) etc
    foo(i)
Run Code Online (Sandbox Code Playgroud)

我的意思是像这样简单的一行

for(auto i: range(10)) //or range(4, 10, 2) or range(0.5, 1.0, 0.1) etc
    foo(i);
Run Code Online (Sandbox Code Playgroud)

但不是这样:

std::vector<int> v(10);
std::iota(begin(v), end(v), 0);
for(auto i: v) {
    foo(i);
}
Run Code Online (Sandbox Code Playgroud)

或这个

for(auto i: []{vector<size_t> v(10); return iota(begin(v), end(v), 0), v;}() ) {
    foo(i);         
}
Run Code Online (Sandbox Code Playgroud)

当然,使用这些例子并不难,或者只是for(;;)我希望有一种方法可以在python中简单而简洁地做到这一点。

min*_*iot 2

类似 Python 的range概念并不是开箱即用的,但您可以Range使用简单的迭代器来创建自己的类,如下所示:

#include <iostream>

template <typename T>
class Range
{
public:
  class iterator
  {
  public:
    explicit iterator(T val, T stop, T step) : m_val(val), m_stop(stop), m_step(step) { }
    iterator& operator ++ ()
    {
      m_val += m_step;
      if ((m_step > 0 && m_val >= m_stop) ||
          (m_step < 0 && m_val <= m_stop))
      {
        m_val = m_stop;
      }
      return *this;
    }
    iterator operator ++ (int) { iterator retval = *this; ++(*this); return retval; }
    bool operator == (iterator other) const {return m_val == other.m_val;}
    bool operator != (iterator other) const {return !(*this == other);}
    T operator * () const { return m_val; }
  private:
    T m_val, m_stop, m_step;
  };

  explicit Range(T stop)
    : m_start(0), m_stop(stop), m_step(1)
  { }

  explicit Range(T start, T stop, T step = 1)
    : m_start(start), m_stop(stop), m_step(step)
  { }

  iterator begin() const { return iterator(m_start, m_stop, m_step); }
  iterator end() const { return iterator(m_stop, m_stop, m_step); }

private:
  T m_start, m_stop, m_step;
};

template <typename T>
Range<T> range(T stop) { return Range<T>(stop); }

template <typename T>
Range<T> range(T start, T stop, T step = 1) { return Range<T>(start, stop, step); }

int main()
{
  for (auto i : range(10)) { std::cout << " " << i; }
  std::cout << std::endl;
  for (auto i : range(4, 10, 2)) { std::cout << " " << i; }
  std::cout << std::endl;
  for (auto i : range(0.5, 1.0, 0.1)) { std::cout << " " << i; }
  std::cout << std::endl;
}
Run Code Online (Sandbox Code Playgroud)

为了支持基于范围for,迭代器类型和begin()/end()函数将完成这项工作。(当然,我上面的实现既快又脏,可能还可以改进。)

您无法像这样滚动自己的类,但是一旦拥有它,其用法就非常类似于 Python 方法:

for (auto i : range(stop)) { ... }
for (auto i : range(start, stop, step)) { ... }
Run Code Online (Sandbox Code Playgroud)

示例输出(请参阅此处的实时版本):

$ g++ -std=c++11 -o test test.cpp && ./test
 0 1 2 3 4 5 6 7 8 9
 4 6 8
 0.5 0.6 0.7 0.8 0.9 1
Run Code Online (Sandbox Code Playgroud)

如果你只需要整数范围,也可以使用boost::irange(感谢Yakk的提醒)。