在 C++ 中设置离散分布

Question

经过几个小时的努力解决这个问题，我找不到任何对我的错误的解释。

我希望计算机选择一个 0 到 120（含）之间的随机数（加权）。我有一个数组，interval[]，它保存从 0 到 120（含）的数字。我有另一个数组，weights[]，它保存选择数组中第 i 个元素的概率。我想为这些数据定义一个概率分布函数。

这是我尝试过的。我收到一条错误消息，指出没有构造函数的实例与参数列表匹配。

我的代码片段：

std::vector< int> weights(121);
for (int i = 0; i < 121; i++)
{
    weights[i] = (teamData[i]).S();
}
discrete_distribution<> dist(weights.begin(), weights.end());

Answer 1

来自您的链接页面（强调我的）

std::piecewise_constant_distribution

产生随机 浮点 数字，均匀分布在每个数中 子区间 [b_i, b_i+1)，每个子区间都有自己的权重 w_i。的集合 区间边界和权重集是这个的参数 分布。

它需要浮点权重和边界，并且权重比边界少一。它也不会输出 0-120 之间的整数，而是浮点数。 您向它传递了整数权重和边界，因此它无法编译。但即使你修复了这个问题，你仍然会从中得到浮点值......

相反，正如您发现的那样，您想要

discrete_distribution

，您可以像这样设置：（从链接页面文档修改）

#include <iostream>
#include <map>
#include <random>
 
int main()
{
    // Setup the random bits
    std::random_device rd;
    std::mt19937 gen(rd());

    // Setup the weights (in this case linearly weighted)
    std::vector<int> weights;
    for(int i=0; i<120; ++i) {
        weights.push_back(i);
    }

    // Create the distribution with those weights
    std::discrete_distribution<> d(weights.begin(), weights.end());

    // use the distribution and print the results.
    std::map<int, int> m;
    for(int n=0; n<10000; ++n) {
        ++m[d(gen)/10];
    }
    for(auto p : m) {
        std::cout << p.first*10 << " - "<<p.first*10+9 << " generated " << p.second << " times\n";
    }
}

Answer 2

 // piecewise_constant_distribution
#include <iostream>
#include <array>
#include <random>

int main()
{
  const int nrolls = 10000; // number of experiments
  const int nstars = 100;   // maximum number of stars to distribute

  std::default_random_engine generator;
  std::array<double,6> intervals {0.0, 2.0, 4.0, 6.0, 8.0, 10.0};
  std::array<double,5> weights {2.0, 1.0, 2.0, 1.0, 2.0};
  std::piecewise_constant_distribution<double>
    distribution (intervals.begin(),intervals.end(),weights.begin());

  int p[10]={};

  for (int i=0; i<nrolls; ++i) {
    int number = distribution(generator);
    ++p[number];
  }

  std::cout << "a piecewise_constant_distribution:" << std::endl;
  for (int i=0; i<10; ++i) {
    std::cout << i << "-" << i+1 << ": ";
    std::cout << std::string(p[i]*nstars/nrolls,'*') << std::endl;
  }
  return 0;
}