c++ - 如何在 C++ 中从具有不同概率的数组中随机选择元素

Question

假设我有一个 vector<Point> p一些对象。

我可以通过简单的 p[rand() % p.size()] 选择一个均匀随机的.

现在假设我有另一个相同大小的 double vector vector <double> chances .

我想从 p 中随机抽样，每个元素的概率类似于 chances 中的值(可能不等于 1.0)。如何在 C++ 中实现这一点？

Answer 1

您正在寻找 std::discrete_distribution。忘记 rand()。

#include <random>
#include <vector>

struct Point {};

int main() {
    std::mt19937 gen(std::random_device{}());

    std::vector<double> chances{1.0, 2.0, 3.0};
    // Initialize to same length.
    std::vector<Point> points(chances.size());
    // size_t is suitable for indexing.
    std::discrete_distribution<std::size_t> d{chances.begin(), chances.end()};

    auto sampled_value = points[d(gen)];
}

为方便起见，权重之和不必为 1。