c++ - 用随机数创建一个 vector

Question

我将如何创建一个充满随机数的 vector ？

通常的代码是这样的:

std::mt19937 rng {std::random_device{}()};
std::uniform_int_distribution<int> dist {1, 52};

std::vector<int> vec(10);
std::generate(begin(vec), end(vec), [&]{return dist(rng);} );

然而，这意味着每个值都被触摸了两次:一次设置为零，然后设置为随机值 (even at O3)

那么如何尽可能高效地做到这一点呢？

Answer 1

您可以创建一个函数调用迭代器并将其传递给 vector 范围构造函数:

#include <boost/iterator/iterator_facade.hpp>
#include <iostream>
#include <vector>
#include <random>
#include <tuple>

template<class F, class Tag = std::input_iterator_tag>
class FunctionCallIterator
    : public boost::iterator_facade<
          FunctionCallIterator<F, Tag>,
          typename std::result_of<F()>::type,
          Tag,
          typename std::result_of<F()>::type
      >
{
    std::tuple<F, ptrdiff_t> m_; // Enable empty base class optimization for empty F.
    friend class boost::iterator_core_access;
    typename std::result_of<F()>::type dereference() const { return std::get<0>(m_)(); }
    bool equal(FunctionCallIterator const& b) const { return std::get<1>(m_) == std::get<1>(b.m_); }
    void increment() { ++std::get<1>(m_); }
    void decrement() { --std::get<1>(m_); }
    void advance(ptrdiff_t n) { std::get<1>(m_) += n; }
    ptrdiff_t distance_to(FunctionCallIterator const& b) const { return std::get<1>(b.m_) - std::get<1>(m_); }
public:
    FunctionCallIterator(F const& f, ptrdiff_t n) : m_(f, n) {}
};

int main() {
    std::mt19937 rng {std::random_device{}()};
    std::uniform_int_distribution<int> dist {1, 52};
    auto f = [&]{return dist(rng);};
    using RngIter = FunctionCallIterator<decltype(f), std::random_access_iterator_tag>;
    std::vector<int> vec(RngIter{f, 0}, RngIter{f, 10});
    for(auto v : vec)
        std::cout << v << '\n';
}

与 push_back/back_inserter 方法相比，此方法不检查当前 vector 大小与其容量的关系，也不增加每个元素的 vector 大小。