蓄水池算法

【题目】有一个机器按自然数序列的方式吐出球（1号球，2号球，3好球。。。），你有一个袋子，袋子最多只能装下K个球，并且除袋子以外，你没有更多的空间。设计一种选择方式，使得当机器吐出第N号球的时候（N>K）,你袋子中的球数是k个，同时可以保证从1号球到N号球中的每一个，被选进袋子的概率都是K/N。举一个更具体的例子，有一个只能装下10个球的袋子，当吐出100个球时，袋子里有10个球，并且1_{100号中的每一个球被选中的概率都是10/100。然后继续吐球，当吐出1000个球时，袋子里有10个球，并且1}1000号中的每一个球被选中的概率都是10/1000.继续吐球，当吐出i个球时，袋子里有10个球，并且1~i号中的每一个球被选中的概率都是10/i,即吐球的同时，已经吐出的球被选中的概率也动态的变化。

蓄水池算法：

1. 处理1~k号球时，直接放进袋子里。
2. 处理第i号球时（i>k），以k/i的概率决定是否将第i号球放进袋子。如果不决定将第i号球放进袋子，直接扔掉第i号球。如果决定将第i号球放进袋子，那么就从袋子里的k个球中随机扔掉一个，然后把第i号球放入袋子。
3. 处理第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;
	}
}