std::ranges::generate_n

< cpp‎ | algorithm‎ | ranges
定义于头文件 <algorithm>
调用签名
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 );
 (C++20 起)

若 0 < n 则对范围 [first, first + n) 中的每个元素赋值连续调用函数对象 gen 的结果，否则不做任何事。

此页面上描述的仿函数实体是 niebloid ，即：

实际上，它们能以函数对象，或以某些特殊编译器扩展实现。

参数

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

返回值

若 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++; });
    // 与 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)

参阅

(C++20)
 保存函数结果到一个范围中
(niebloid)
(C++20)
 将一个给定值复制赋值给一个范围内的每个元素
(niebloid)
(C++20)
 将一个值复制赋值给一定量的元素
(niebloid)
(C++20)
 将一个函数应用于某一范围的各个元素
(niebloid)
将相继的函数调用结果赋值给一个范围中的 N 个元素
(函数模板)