c++ - 用 t 个随机数生成位 vector

Question

我想生成一个数组/vector v，大小为 p，其中 t 为一，p-t 为零。 t 的位置必须是随机的。

这是我迄今为止编写的解决方案，但我不确定它是否是最有效的解决方案。此外，我以前从未使用过 random_device 或 mt19937 (发现它们 here )，所以我不知道有什么可能的缺点。

#include <algorithm>
#include <random>
#include <vector>
...
int p=10, t=3;
std::vector<int> v(p,0);
for(int i=0;i<t;i++)  //better way?
    v[i] = 1;
std::random_device rd;
std::mt19937 g(rd());
std::shuffle(std::begin(v), std::end(v), g);

这是我要重现的 matlab 代码:

rp = randperm(p);
I_s(i,:) = rp(1:t);
v = zeros(p,1);
v(I_s(i,:)) = 1;

Demo here!

Answer 1

std::shuffle(std::begin(v), std::end(v), g);

不会使位位置随机，它只是随机更改 vector 中现有项(它们都设置了零位置位)的位置。

---

如果我正确理解了您的要求，您实际上应该拥有的是:

std::random_device rd;
std::mt19937 g(rd());
for(int i=0;i<t;i++)
     v[i] = 1 << rd(); // set a random bit position