等概率随机取数算法的几种实现(洗牌算法)

等概率随机取数算法的几种实现

  最近读了项目中的工具脚本,发现一个随机取数的函数,功能大概是从M个数中不重复的随机取出N个数,算是数组随机排序然后取前N个值的变种。

  脚本实现采取原始的方法,每随机取一个数就放到一个数组中,下次取数时遍历结果数组判断是否已经取出,平均时间复杂度为O(MlogM),空间复杂度O(N),效率不高。

  想了一下解决方案,能优化的地方应该就是将遍历数组判断是否取出使用哈希或者红黑树实现,以空间换时间,虽然可以降低时间复杂度,但原算法仍存在问题:当M接近于N的时候,效率会急剧下降,十分恐怖。

  如果借助洗牌算法,获取一个随机排列的子集,便能实现等概率随机取数的功能。

1. Fisher-Yates Shuffle算法

  最早于1938年由Ronald Fisher和Frank Yates所著《Statistical tables for biological,agricultural and medical research》提出,算法描述为:

  1. 将1到N数字存到数组中
  2. 从数组中取一个1到剩下数字个数的随机数k
  3. 从低位开始,将数组第k个数字取出,并保存到结果数组末尾
  4. 重复第2步,直到所有数字都被取出
  5. 第3步得到的结果数组就是所求的随机序列

该算法存在对数组随机元素的删除操作,时间复杂度O(N^2),空间复杂度O(M),效率仍然比较低。

  • 1
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值