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

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

 

 

解法：

 

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

 

设getNum取的第i个数为numi，numi存入num中的概率为Pi。设Pi=1/i ( i已知时，所以这个概率很容易做到)。

则numi保留，并最后返回的情况是，numi被存入num中，且numj(i<j<=N)不被存入num。其概率为

P(numi)=P_i*(1- P_i+1) * (1 - P_i+2)* ...(1 - P_N)=1/i * (i/(i+1)) * ((i+1)/(i+20)*....((N-1)/N) = 1/N