洗牌算法是将原来的数组打散,使原数组的某个数在打散后的数组的每个位置等概率出现。
由抽牌、换牌、插牌衍生出三种算法。
抽牌算法
每次在原数组里随机取一个数,直到把所有的数取完。
// 从原始数组中随机取一个之前没取过的数字到新数组
function shuffle(arr) {
let new_arr = [];
while(arr.length > 0) {
let index = Math.floor(Math.random() * arr.length);
new_arr.push(arr[index]);
arr.splice(index, 1);
}
return new_arr;
}
// 验证各个元素被放在新数组的某个位置的概率是相等的
let arr = [1, 2, 3, 4, 5, 6, 7];
let count = [];
for(let i = 0; i < arr.length; i++) {
// count[i] = Array(arr.length).fill(0);
count[i] = new Map;
}
for(let i = 0; i < 10000; i++) {
arr = shuffle(arr);
for(let i = 0; i < arr.length; i++) {
if(count[i].has(arr[i])) {
count[i].set(arr[i], count[i].get(arr[i]) + 1);
}
else {
count[i].set(arr[i], 1);
}
}
}
console.log(count);
这里我先用了统计学的验证,即洗一万次牌,看每张牌出现在任意位置的几率是否相近,
可以看出基本是维持在一千三到一千四左右的。
10000
/
7
≈
1428
10000 / 7 \approx1428
10000/7≈1428
接下来从数学角度验证
证明:任意一个元素被放入第i个位置的概率P = 前i个位置没有选中m的概率 * 第i个位置选中的概率
P
=
n
−
1
n
×
n
−
2
n
−
1
×
.
.
.
×
n
−
i
+
1
n
−
i
+
2
×
1
n
−
i
+
1
=
1
n
P = \frac{n-1}{n} \times \frac{n-2}{n-1} \times ... \times\frac{n-i+1}{n-i+2}\times\frac{1}{n-i+1} = \frac{1}{n}
P=nn−1×n−1n−2×...×n−i+2n−i+1×n−i+11=n1
换牌算法
从第一个元素开始,与剩余元素交换(0, length -1),然后是第二个元素(1, length -1)。直到交换到直剩最后一个元素没办法交换为止。
// 换牌算法
function shuffle2(arr) {
for(let i = 0; i < arr.length - 1; i++) {
let index = Math.floor(Math.random() * (arr.length -i)) + i;
[arr[i], arr[index]] = [arr[index], arr[i]];
}
return arr;
}
与抽牌的区别:
时间复杂度都是O(n)
但是空间复杂度,抽牌是O(n), 换牌算法是O(1)就地,换牌要更好。