从JS中的大小为n的数组中随机选取m个整数

时间:2016-01-13 02:07:18

标签: javascript java random

问题(来自破解编码面试):编写一个方法,从大小为n的数组中随机生成一组m个整数。 每个元素必须具有相同的被选择概率。

我在JS中实现了我的答案。对于递归函数,代码有时会返回undefined作为数组中的一个元素。

我的JS代码

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;
};

给定Java解决方案

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;
}

1 个答案:

答案 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;的数组中的数据在我看来非常重要。