Underscore源码阅读:shuffle,sample

shuffle

Fisher-Yates shuffle;官方写了shuffle使用这种算法在进行随机乱序。

不过说真的,我没看懂源码是怎么实现的;尽管我看懂了Fisher-Yates算法,但是我怎么觉得underscore源码的实现跟这个算法讲得不一样……

这里写下洗牌算法的思路吧。
简单讲,假设我们有一个数组,[1, 2, ,3, 4, 5, 6],现在我们要对这个数组进行随机乱序。洗牌算法的思路就是,我们用两个数组,一个代表还没有被选择的元素数组A,一个代表已经被选择了的元素数组B。

即一开始的两个数组的内容是:[1, 2, 3, ,4, 5, 6][]

洗牌开始,我们从数组A中随机选择一个元素,random(0, 5)随机生成一个0~5的数组下标;然后我们将这个元素推入数组B。假设random(0, 5)生成了一个2,那么我们就要B.push(A[2])

这样得到的数组内容是:[1, 2, 4, 5, 6] [3]

这样一直洗下去;注意从第二轮开始,随机数生成的范围就变成了random(0, index--)

这大致就是算法的内容。当然我们也看到了,如果完全按照算法写的,我们会用到大量的splice。优化的方案就是不再使用两个数组,而使用一个数组,用数组的左半部分表示没有被选择的元素,用数组的右半部分表示被选择过的元素。

即如下一个顺序:

  1. 初始数组:array = [1, 2, 3, 4, 5, 6]
  2. 随机生成一个数组下标(范围仍然是0 ~ 5),假设生成2,我们希望将array[2]弄到整个数组的右边
  3. 交换被选中的元素和左半数组的最后一个元素,即交换3和6
  4. array = [1, 2, 6, 4, 5, 3];

这样就完成了一轮洗牌,继续洗下去就好了。

代码实现

var random = function (min, max) {
  return min + Math.floor(Math.random() * (max - min + 1))
}

var shuffle = function (array) {
  var res = array.slice();
  for (var len = res.length - 1; len > 0; len--) {
    var index = random(0, len), temp = res[index];
    res[index] = res[len];
    res[len] = temp;
  }
  console.log(res);
  return res;
}

var array = [1, 2, 3, 4, 5, 6];
shuffle(array);

sample(list, [n])

从 list中产生一个随机样本。传递一个数字表示从list中返回n个随机元素
sample的基本思路就是就是通过shuffle生成一个随机乱序,然后要多少个元素,就返回多少个元素

var sample = function (list, n) {
  n = n || 1;
  return shuffle(list).slice(0, n);
}

你可能感兴趣的:(Underscore源码阅读:shuffle,sample)