在C++/STL中是否有一个与Python range()相当的紧凑型

Question

如何使用C++/STL执行以下等效操作？我想填充std::vector一系列值[min,max].

# Python
>>> x = range(0, 10)
>>> x
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

我想我可以使用std::generate_n并提供一个函子来生成序列,但我想知道是否有更简洁的方法使用STL这样做？

Answer 1

在C++ 11中,有std::iota:

std::vector<int> x(10);
std::iota(std::begin(x), std::end(x), 0); //0 is the starting number

Answer 2

有boost :: irange:

std::vector<int> x;
boost::push_back(x, boost::irange(0, 10));

Answer 3

我最终编写了一些实用函数来做到这一点。您可以按如下方式使用它们：

auto x = range(10); // [0, ..., 9]
auto y = range(2, 20); // [2, ..., 19]
auto z = range(10, 2, -2); // [10, 8, 6, 4]

代码：

#include <vector>
#include <stdexcept>

template <typename IntType>
std::vector<IntType> range(IntType start, IntType stop, IntType step)
{
  if (step == IntType(0))
  {
    throw std::invalid_argument("step for range must be non-zero");
  }

  std::vector<IntType> result;
  IntType i = start;
  while ((step > 0) ? (i < stop) : (i > stop))
  {
    result.push_back(i);
    i += step;
  }

  return result;
}

template <typename IntType>
std::vector<IntType> range(IntType start, IntType stop)
{
  return range(start, stop, IntType(1));
}

template <typename IntType>
std::vector<IntType> range(IntType stop)
{
  return range(IntType(0), stop, IntType(1));
}

Answer 4

多年来我一直在使用这个库来实现这个目的：

https://github.com/klmr/cpp11-range

效果很好，并且代理已被优化。

for (auto i : range(1, 5))
    cout << i << "\n";

for (auto u : range(0u))
    if (u == 3u) 
        break;
    else         
        cout << u << "\n";

for (auto c : range('a', 'd'))
    cout << c << "\n";

for (auto i : range(100).step(-3))
    if (i < 90) 
        break;
    else        
        cout << i << "\n";

for (auto i : indices({"foo", "bar"}))
    cout << i << '\n';