您的位置:首页 > 职场人生

一道面试题,据说是阿里巴巴的。

2010-03-31 13:56 281 查看
有一个函数int getNum(),每运行一次可以从一个数组V
里面取出一个数,N未知,当数取完的时候,函数返回NULL。现在要求写一个函数int get(),这个函数运行一次可以从V
里随机取出一个数,而这个数必须是符合1/N平均分布的,也就是说V
里面任意一个数都有1/N的机会被取出,要求空间复杂度为O(1)(http://topic.csdn.net/u/20091024/12/989417AA-60E9-45D1-A96F-A623695FC6D7.html)

解法:

设置一个整形num,用来存放get()返回的数。调用getNum()从V
取数,按一定概率存入num中(覆盖以前的),直到getNum()返回NULL。

设getNum取的第i个数为numi,numi存入num中的概率为Pi。设Pi=1/i ( i已知时,所以这个概率很容易做到)。
则numi保留,并最后返回的情况是,numi被存入num中,且numj(i<j<=N)不被存入num。其概率为
P(numi)=Pi*(1- Pi+1) * (1 - Pi+2)* ...(1 - PN)=1/i * (i/(i+1)) * ((i+1)/(i+20)*....((N-1)/N) = 1/N
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: