我们考虑如何打乱一个数组:
打乱的含义是:等概率的产生所有的排列情况,即:每个元素出现在每个位置的概率相等。因此,算法的步骤为:
循环n次,在第i次循环中(0≤i<n):
- 在[i,n)中随机抽取一个下标j;
- 将第i个元素与第j个元素交换
其中数组中的 nums[i … n-1]的部分为待乱序的数组,其长度为 n−i;nums[0 … i−1] 的部分为乱序后的数组,其长度为 i。
证明也很简单:想要证明洗牌算法是等概率的,只要看每个元素出现的概率是不是1/n即可。
对于第一个位置,随机从n个元素中挑选一个,元素出现的概率显然是1/n。对于第二个位置,这个元素没有出现在第一个位置概率为(n-1)/n,从剩下的n-1个元素中挑选一个元素概率为1/(n-1),因此概率为:((n-1)/n)*(1/(n-1))=1/n;同理,每个位置任意元素出现的概率都为1/n。代码如下:
vector<int> shuffle() {
for (int i = 0; i < n; i++) {
int tmp;
int target = i + rand() % (n - i);
tmp = current[i];
current[i] = current[target];
current[target] = tmp;
}
return current;
}
事实上:python3中具有直接修改原序列,将序列中的元素随机打乱的函数:random.shuffle() 。