std::ranges::generate_n

template< std::input_or_output_iterator O, std::copy_constructible F >
requires std::invocable<F&> && std::indirectly_writable<O, std::invoke_result_t<F&>>
constexpr O
    generate_n( O first, std::iter_difference_t<O> n, F gen );

Присваивает результат последовательных вызовов функционального объекта gen каждому элементу в диапазоне [first, first + n), если 0 < n. В противном случае ничего не делает.

Функциональноподобные сущности, описанные на этой странице, являются объектами-функциями алгоритмов (неофициально известные как niebloids), а именно:

• Явные списки аргументов шаблона не могут быть указаны при вызове любой из них.
• Ни одна из них не видна для поиска, зависящего от аргументов.
• Когда любая из них найдена путем обычного неквалифицированного поиска как имя слева от оператора вызова функции, поиск, зависящий от аргументов подавляется.

Параметры

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

Итератор на элемент, следующий за последним присвоенным, если 0 < count , first в противном случае.

Сложность

Ровно n вызовов gen ( ) и присваиваний.

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

struct generate_n_fn
{
    template<std::input_or_output_iterator O, std::copy_constructible F>
    requires std::invocable<F&> && std::indirectly_writable<O, std::invoke_result_t<F&>>
    constexpr O operator()(O first, std::iter_difference_t<O> n, F gen) const
    {
        for (; n-- > 0; *first = std::invoke(gen), ++first)
        {}
        return first;
    }
};
inline constexpr generate_n_fn generate_n {};

Пример

#include <algorithm>
#include <array>
#include <iostream>
#include <random>
#include <string_view>
auto dice()
{
    static std::uniform_int_distribution<int> distr {1, 6};
    static std::random_device engine;
    static std::mt19937 noise {engine()};
    return distr(noise);
}
void print(const auto& v, std::string_view comment)
{
    for (int i : v)
        std::cout << i << ' ';
    std::cout << '(' << comment << ")\n";
}
int main()
{
    std::array<int, 8> v;
    std::ranges::generate_n(v.begin(), v.size(), dice);
    print(v, "dice");
    std::ranges::generate_n(v.begin(), v.size(), [n {0}] mutable { return n++; });
    // same effect as std::iota(v.begin(), v.end(), 0);
    print(v, "iota");
}

5 5 2 2 6 6 3 5 (dice)
0 1 2 3 4 5 6 7 (iota)

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