储水池抽样算法原理与实现

原创作品，出自 “晓风残月xj” 博客，欢迎转载。转载时请务必注明出处（http://blog.csdn.net/xiaofengcanyuexj）。

因为各种原因，可能存在诸多不足，欢迎斧正！

记得在实习那会，涉及到抽奖的解决方式，即一天以k/n的概率中奖，要求给出简单高效的算法。当中n仅仅有在结束时才知道。

换言之，题意即为怎样从未知或者非常大样本n空间中随机地取k个数，保证每一个被取到的概率为n/k。解决此题的非常好方法就是储水池算法Reservoir
Sampling Algorithm。

储水池算法应用的前提条件是样本空间总数未知或非常大。若当前的样本数为m。则以k/m的概率保留每个样本……直到终于样本数为n时是的每个被选中的概率为k/n，缺陷是不到最后不知道哪些样本要被选中。这样问题就来了，为何不在终于结果n出来时再来随机抽取k个样本，保证概率为k/n呢？事实上这样的想法有些时候是不可行的，数据是动态增长的。可能缓存系统存储不了全部的样本信息。但却足够存储k个样本的信息，即空间复杂度能够从O(n)降到O(k)，k相对n来说是非常小的。

所以储水池算法还是非常有应用价值的。

以下说说储水池算法的原理及实现。先把前k个数放入蓄水池；对第k+1个样本，以k/(k+1)概率决定换入蓄水池，换入时又随机选取池里的一个样本作为替换项……，相应第i个样本（i>k）,以k/i的概率留下。然后随机从池里选中一个替换掉……这样一直做下去，对于随意的样本空间n，对单个样本的选取概率为k/n。

证明步骤例如以下：



此算法时间复杂度为O(n)，空间复杂度为O(k)，可以处理样本空间总数未知或非常大的情况。不失为一种比較好的方法。关于储水池算法。在分布式系统上有着应用，在此提供一下丰富眼界的链接。储水池算法的拓展。

因为工作原因。Java对于我来说似乎比C/C++更重要一点，全部以后的代码尽量用Java给出，直接上源码的。

import java.util.Random;

public class ReservoirSampling {

private int m_nChoice;
private int [] m_arrayChoice;
private int m_nNowNumber;
private Random m_Rand =new Random(System.currentTimeMillis());

public ReservoirSampling(int k)
{
m_nNowNumber = 0;
m_nChoice = k;
m_arrayChoice = new int [k];
}

public void AdjustChoice()
{
if(m_nNowNumber < m_nChoice)
{
m_arrayChoice [m_nNowNumber] = m_nNowNumber;
}
else
{
int id = m_Rand.nextInt(m_nNowNumber + 1);
if(id < m_nChoice)
{
m_arrayChoice [id] = m_nNowNumber;
}
}

++ m_nNowNumber;
}

public void ShowChoice()
{
for( int i = 0; i <m_arrayChoice.length ; ++ i)
{
if(i != 0 && i % 10 == 0)
System.out.println();
System.out.print(1 + m_arrayChoice [i] + "  ");
}
}

public static void main(String agr[])
{
System.out.println("My Java program!");
ReservoirSampling test = new ReservoirSampling(20);
for( int i =0 ;i < 200 ; ++ i)
{
test.AdjustChoice();
}
test.ShowChoice();
}

}

因为时间和水平的原因，难免有不足的地方。欢迎斧正！