[英]Pick m integers randomly from an array of size n in JS
问题 (来自破解编码面试 ):编写一种方法,从大小为n的数组中随机生成一组m个整数。 每个元素必须具有相等的被选择概率。
我正在用JS实现我的答案。 对于递归函数,代码有时返回undefined作为数组中的元素之一。
var pickMRecursively = function(A, m, i) {
if (i === undefined) return pickMRecursively(A, m, A.length);
if (i+1 === m) return A.slice(0, m);
if (i + m > m) {
var subset = pickMRecursively(A, m, i-1);
var k = rand(0, i);
if (k < m) subset[k] = A[i];
return subset;
}
return null;
};
int[] pickMRecursively(int[] original, int m,int i) {
if (i +1==m){// Basecase
/* return first m elements of original */
} elseif(i+m>m){
int[] subset = pickMRecursively(original, m, i - 1);
int k = random value between 0 and i, inclusive
if(k<m){
subset[k] = original[i]j
}
return subset;
}
return null;
}
我讨厌这些问题,因为有时它们通常是故意含糊的-我会问“数组中的数据类型是什么?”。 但是,如果这实际上是关于随机重新排列数组的问题,那么在JavaScript中,假设arr
是一个数字数组,其中某些/全部可能不是整数...
function generateM(arr) {
var hold = [];
var m = [];
var n = arr.length;
var grab;
// clone arr >> hold
while(n--) {
hold[n] = arr[n];
}
n = hold.length;
// select randomly from hold
while(n--) {
grab = hold.splice(Math.floor(Math.random()*n),1)[0];
// ensure integers
m.push(Math.round(grab));
}
return m;
}
在此处克隆了数组arr
,以涵盖范围界定问题并产生新的集合,而不是对现有集合进行重新排序。
ADDIT :或者,如果这只是要求从n.length
数组生成的一组m.length
随机整数,则该数组的实际内容和可能的范围(随机生成)都无关紧要值将是(可能是?) 0 - n.length
,所以...
function generateM(arr, M) {
var aLen = arr.length;
var m = [];
do {
m.push(Math.round(Math.random() * aLen));
} while(M--);
return m;
}
...但这似乎是一个愚蠢,毫无意义的挑战。 在我看来,“大小为n的数组”中的数据非常重要。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.