leetcode_middle_38_398. Random Pick Index

题意:

给出一个数,在数组中查找这个数,并随机等概率返回一个这个数的角标。

分析:

这个题在leetcode中属于一类题型,比较优的解法我认为比较简单但是是比较难想的。但是由于我已经做过一道这样的题,知道方法,就没障碍了。回顾一下这个概率方法:

我们要在一列n个数中选一个数,要求每一个数被选中的概率都是1/n,由于我们不想存储,有什么办法一边遍历一边确定呢?

那么一定是出来一个就要选一次,比如5个数,出来3个数了,就已经选了3次了,第4个数还要选。那么概率是什么呢?

比如第四个数被选的时候概率是p4,第5个数被选的时候概率是p5,我们知道一个数如果后面还要选择(扩大样本范围,概率是会被后面所影响的)。

但是第5个数是最后一个数,后面没有了,概率就不会被影响,所以概率必然是1/5,好了,那么第四个呢?就是p4*(1 - 1/5),意思就是第五次没有选中第五个(也即是选中了前四个)再乘上p4(前四个里面选中p4的概率)就是最终选定第4个的概率,而由于这个概率是1/5,我们解等式p4*(1-1/5) = 1/5   =>  p4=1/4,同理可知顺序遍历动态选择第n个的时候概率应该是1/n.

代码:

public class Solution {
    int[] nums;
    Random random;
    
    public Solution(int[] nums) {
        this.nums = nums;
        random = new Random();
    }
    
    public int pick(int target) {
        int r = 0;
        for(int i=0,j=1; i<nums.length; i++){   //j代表第j次选择
            if(nums[i] == target){
                if(random.nextInt(j) == 0){  //if(1/j的概率)
                    r = i;
                }
                j++;
            }   
        }
        return r;
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值