【题目】有一个机器按自然数序列的方式吐出球(1号球,2号球,3好球。。。),你有一个袋子,袋子最多只能装下K个球,并且除袋子以外,你没有更多的空间。设计一种选择方式,使得当机器吐出第N号球的时候(N>K),你袋子中的球数是k个,同时可以保证从1号球到N号球中的每一个,被选进袋子的概率都是K/N。举一个更具体的例子,有一个只能装下10个球的袋子,当吐出100个球时,袋子里有10个球,并且1100号中的每一个球被选中的概率都是10/100。然后继续吐球,当吐出1000个球时,袋子里有10个球,并且11000号中的每一个球被选中的概率都是10/1000.继续吐球,当吐出i个球时,袋子里有10个球,并且1~i号中的每一个球被选中的概率都是10/i,即吐球的同时,已经吐出的球被选中的概率也动态的变化。
蓄水池算法:
- 处理1~k号球时,直接放进袋子里。
- 处理第i号球时(i>k),以k/i的概率决定是否将第i号球放进袋子。如果不决定将第i号球放进袋子,直接扔掉第i号球。如果决定将第i号球放进袋子,那么就从袋子里的k个球中随机扔掉一个,然后把第i号球放入袋子。
- 处理第i+1号球时重复步骤1或步骤2.
public class XuShuiChi {
public static int rand(int max) {
return (int) (Math.random() * max) + 1;
}
public static int[] getKNum(int k, int max) {
if (k < 1 || max < 1) {
return null;
}
int[] res = new int[Math.min(max, k)];
for (int i = 0; i != res.length; i++) {
res[i] = i + 1;
}
for (int i = k + 1; i < max +1; i++) {
if (rand(i) <= k) {
res[rand(k) - 1] = i;
}
}
return res;
}
}