蓄水池算法

// WaterPool.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <stdlib.h>
#include <vector>
#include <iterator>
#include <iostream>
using namespace std;


int GetRand(int i)
{
    return ((int)rand()/(int)RAND_MAX)*i + 1;
}

int _tmain(int argc, _TCHAR* argv[])
{//数据流为1~1000，池子大小100，等概率地从中任意选出100个数存入池子中
    int index = 0;
    int maxFlow = 1000;//连续的流，最大值已知
    int poolSize = 100;
    vector<int> pool;


    for(;index < poolSize;index++)
        pool.push_back(index+1);
//1.1~100依次取进来
    for(index = poolSize;index < maxFlow;index++)
    {
        if(GetRand(index) <= poolSize)
            pool[GetRand(poolSize) - 1] = index;
    }
//2.100以后的 根据索引产生一个随机数，随机数在0~100之间就放入池子中，
//然后随机从池中替换掉一个。最终每个数进池的概率都为poolSize/maxFlow
    vector<int>::iterator ite =  pool.begin();
    for(;ite != pool.end();ite++)
        cout<<*ite<<" ";
    cout<<endl;
    system("pause");
    return 0;
}