使用 std::binomial_distribution 在范围之间生成随机数

Generate random numbers between range using std::binomial_distribution

本文关键字：之间 范围 随机数 distribution std binomial 使用      更新时间：2023-10-16

我创建了一个函数来生成稀疏矩阵，但是随机生成器只生成1 的值，我需要让它生成特定范围内的数字。

size_t spare_matrix_generator(int *M, size_t rows, size_t cols)
{
std::random_device rd;
std::mt19937 gen(rd());
gen.seed(time(NULL));
std::binomial_distribution<int> distribution(1, 0.5);
size_t NNZ = 0;
for (szie_t i = 0; i < rows*cols; i++)
{
M[i] = distribution(gen);
std::cout << M[i] << " ";
if (A[i] != 0)
NNZ++;
}
std::cout << std::endl;
return NNZ;
}

假设5x5矩阵，上述函数的示例输出如下所示：

0 1 1 0 0 
1 0 0 0 1
0 0 0 1 0
1 0 0 0 1
0 0 0 0 1

对于 [a， b] 中的数字， 

size_t spare_matrix_generator(int *M, size_t rows, size_t cols)
{
std::random_device rd;
std::mt19937 gen(rd());
gen.seed(time(NULL));
int b = 12, a = 3;
std::binomial_distribution<int> distribution(b-a, 0.5);
// To handle matrix sparsity
float zero_prob = 0.8;
std::mt19937 sparse_gen(rd() + 42);
std::uniform_real_distribution<double> sparse_dist(0.0,1.0);
size_t NNZ = 0;
for (szie_t i = 0; i < rows*cols; i++)
{
if(sparse_dist(sparse_gen) > zero_prob){
M[i] = a + distribution(gen);
} else {
M[i] = 0;
}
std::cout << M[i] << " ";
if (M[i] != 0)
NNZ++;
}
std::cout << std::endl;
return NNZ;
}

相关文章：