数组中的N个随机元素
从数组中获取n
个随机元素,这些元素的键是唯一的,且不超过数组的大小。
- 使用Fisher-Yates算法对数组进行洗牌。
- 使用
Array.prototype.slice()
获取前n
个元素。 - 如果省略第二个参数
n
,则只获取一个随机元素。
const sampleSize = ([...arr], n = 1) => {
let m = arr.length;
while (m) {
const i = Math.floor(Math.random() * m--);
[arr[m], arr[i]] = [arr[i], arr[m]];
}
return arr.slice(0, n);
};
sampleSize([1, 2, 3], 2); // [3, 1]
sampleSize([1, 2, 3], 4); // [2, 3, 1]