std::ranges:: iota, std::ranges:: iota_result

Заполняет диапазон [ first , last ) последовательно возрастающими значениями, начиная с value и повторно вычисляя ++ value .

Эквивалентная операция:

*(first)     = value;
*(first + 1) = ++value;
*(first + 2) = ++value;
*(first + 3) = ++value;
...

Параметры

Возвращаемое значение

{ last, value + ranges:: distance ( first, last ) }

Сложность

Ровно last - first инкрементов и присваиваний.

Возможная реализация

struct iota_fn
{
    template<std::input_or_output_iterator O, std::sentinel_for<O> S,
            std::weakly_incrementable T>
    requires std::indirectly_writable<O, const T&>
    constexpr iota_result<O, T> operator()(O first, S last, T value) const
    {
        while (first != last)
        {
            *first = as_const(value);
            ++first;
            ++value;
        }
        return {std::move(first), std::move(value)};
    }
    template<std::weakly_incrementable T, std::ranges::output_range<const T&> R>
    constexpr iota_result<std::ranges::borrowed_iterator_t<R>, T>
    operator()(R&& r, T value) const
    {
        return (*this)(std::ranges::begin(r), std::ranges::end(r), std::move(value));
    }
};
inline constexpr iota_fn iota;

Примечания

Функция названа в честь целочисленной функции ⍳ из языка программирования APL .

Пример

#include <algorithm>
#include <functional>
#include <iostream>
#include <list>
#include <numeric>
#include <random>
#include <vector>
template <typename Proj = std::identity>
void println(auto comment, std::ranges::input_range auto&& range, Proj proj = {})
{
    for (std::cout << comment; auto const &element : range)
        std::cout << proj(element) << ' ';
    std::cout << '\n';
}
int main()
{
    std::list<int> list(8);
    // Заполняем список возрастающими значениями: 0, 1, 2, ..., 7
    std::ranges::iota(list, 0);
    println("List: ", list);
    // Вектор итераторов (см. комментарий к Примеру)
    std::vector<std::list<int>::iterator> vec(list.size());
    // Заполняем итераторами на последовательные элементы списка
    std::ranges::iota(vec.begin(), vec.end(), list.begin());
    std::ranges::shuffle(vec, std::mt19937 {std::random_device {}()});
    println("List viewed via vector: ", vec, [](auto it) { return *it; });
}

List: 0 1 2 3 4 5 6 7
List viewed via vector: 5 7 6 0 1 3 4 2

Смотрите также