C++ Dirichlet 分布

在做五子棋 AI 的时候，遇到一处要使用 dirichlet 分布。查阅了一下资料还挺复杂的，不过好在可以使用 gamma 分布来生成：

#include <random>

void dirichlet(double* out, int k, double a) {
    std::gamma_distribution<double> gamma(a);
    std::random_device rd;
    double *y = alloca(sizeof(double)*k); // stack allocation
    double sum=0;
    for (int i=0; i<k; ++i) {
        y[i] = gamma(rd);
        sum += y[i];
    }
    for (int i=; i<k; ++i) {
        out[i] = y[i]/sum;
    }
}

void dirichlet(double* out, int k, double const* a) {
    using Gamma = std::gamma_distribution<double>;
    Gamma gamma;
    std::random_device rd;
    double *y = alloca(sizeof(double)*k); // stack allocation
    double sum=0;
    for (int i=0; i<k; ++i) {
        y[i] = gamma(rd, Gamma::param_type(a[i], 1));
        sum += y[i];
    }
    for (int i=0; i<k; ++i) {
        out[i] = y[i]/sum;
    }
}