双层桶-中位数求解

fstream操作参考：http://www.cppblog.com/saga/archive/2007/06/19/26652.html

题目描述：

101个数（[0,1000)之间的数）用双层桶的方法来计算中位数

分析：

1. 分成4个桶即[0,250),[250,500),[500,750),[750,1000)，对数据一次遍历，得到四个桶的内容，分别写入文件，以及每个桶的数据量。

2. 假设用b[0-3]存储，若b[0] = 20,b[1] = 41,b[2] = 20, b[3] = 20，我们知道101个数的中位数是第51大的数据。

3. 假设101个数存储在a[101]中，那么可以通过sort(a,a+101)，a[50]来得到中位数。

tips(对文件操作，若每次都想从新覆盖文件，需加上ios::trunc，第一次创建文件ios::app，以追加新建方式创建文件，第二次就将其去掉)

#include <iostream>
#include <time.h>
#include <fstream>
#include <conio.h>
#include <algorithm>

using namespace std;
// 存放需要找的数据
int a[101];
// 存储3个桶的大小，初始化为0
int b[4];
// 初始化a数组,101个数字，中间值应该是50大的值，最大为1000，最小为0
void initialArray()
{
	srand(unsigned(time(NULL)));
	for(int i=0;i<101;++i)
		a[i] = int((double)rand()/RAND_MAX*1000);
	sort(a,a+101);
	// 输出第50大的值
	cout<<a[50]<<endl;
	/*for(int i=0;i<101;++i)
		cout<<a[i]<<" ";
	cout<<endl;*/
}
void processBucket()
{
	// 四个桶
	fstream arr[4];
	int i;
	memset(b,0,sizeof(b));
	// 初始化读入文件
	for(i=0;i<4;++i)
	{
		char temp[100];
		sprintf_s(temp,99,"d:/test/ccc%d.dat",i);
		arr[i].open(temp,ios::in|ios::out|ios::binary|ios::trunc);
		if(!arr[i])
			cout<<"openfile"<<i<<" error!"<<endl;
	}
	// 将数据划分到桶，分别对每个桶进行统计
	for(int i=0;i<101;++i)
	{
		int j = a[i] / 250;// 划分[0,250),[250,500)....为桶
		b[j]++;
		arr[j].write((char*)&a[i],sizeof(int));
	}
	// 判断到达所需数量时，计算中位数在最后一个桶的位置
	int sum =0,presum=0;
	for(i=0;i<4;++i)
	{
		sum += b[i];
		if(sum >= 51)
			break;
		presum = sum;
	}
	// 第51个元素为
	int v = 51 - presum - 1;
	int k = i;
	cout<<"b（0-3） "<<b[0]<<" "<<b[1]<<" "<<b[2]<<" "<<b[3]<<endl;
	cout<<"sum(b0-bk)="<<sum<<" b[k]="<<b[k]<<endl;
	cout<<"第k"<<k<<"个桶"<<" 第"<<v+1<<"个元素"<<endl;
	
	for(i=0;i<4;++i)
		arr[i].close();

	int colin[100],xx;
	char pathss[100];
	sprintf_s(pathss,99,"d:/test/ccc%d.dat",k);
	fstream only(pathss,ios::in|ios::out|ios::binary);
	only.seekg(0);
	i = 0;
	while(only.read((char*)&xx,sizeof(int)))
	{
		colin[i++] = xx;
	}
	only.close();

	sort(colin,colin+i);
	cout<<colin[v]<<endl;
}
int main()
{
	initialArray();
	processBucket();
	getch();
	return 0;
}