std::ranges::generate_n() 算法

自 C++20 起

简化

// (1)
constexpr O generate_n( O first, std::iter_difference_t<O> n, F gen );

参数类型是泛型的，并具有以下约束：

• O - std::input_or_output_iterator
• F - std::copy_constructible

此外，每个重载都有以下约束

• (1) - invocable<F&> && indirectly_writable<O, invoke_result_t<F&>>

（为方便阅读，此处省略了 std:: 命名空间）

详细

// (1)
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 );

如果 0 < n，则将函数对象 gen 的连续调用结果分配给范围 [first; first + n) 中的每个元素。
否则不执行任何操作。

本页描述的函数类实体是niebloids。

参数

first	要修改的元素范围的起始。
n	要修改的元素数量。
gen	生成器函数对象。

返回值

如果 0 < count，则返回分配的最后一个元素之后一个位置的迭代器；否则返回 first。

复杂度

恰好调用 gen() 并赋值 n 次。

异常

(无)

可能的实现

generate_n(1)

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 {};

示例

Main.cpp

#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)