问题(来自破解编码面试):编写一个方法,从大小为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;
}
答案 0 :(得分:1)
我讨厌这些问题,因为有时候他们经常故意模糊 - 我会问&#34;阵列中有什么类型的数据?&#34;。但是,如果这实际上是一个关于随机重新排序数组的问题,那么在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 :或者,如果这只是要求从m.length
数组生成的一组n.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&#39;的数组中的数据在我看来非常重要。