算法导论CLRS 9.2 选择第i小的元素

/*9.2 选择第i小的元素
 *RANDOMIZED-SELECT
 */
#include<iostream>
#include<iomanip>
#include<vector>
#include<ctime>
using namespace std;
typedef vector<double>::iterator dvecIte;

dvecIte partition(dvecIte iteB, dvecIte iteE)
{//在返回前,ite1指向不小于*(iteE-1)的队列的最左边的元素

	dvecIte ite1 = iteB, ite2;
	for(ite2 = iteB; iteE-1 != ite2; ++ite2) {
		if( *ite2 < *(iteE-1)) {
			if( ite1!= ite2) {
				double temp = *ite1;
				*ite1 = *ite2;
				*ite2 = temp;
			}
			++ite1;
		}
	}
	//如果ite1为iteE-1,说明iteE-1指向最大值,要将其排出
	//而且两者相等时不能用^
	if(ite1 != iteE-1)
	{
		double temp = *ite1;
		*ite1 = *(iteE - 1);
		*(iteE - 1) = temp;
	}
	return ite1;//返回*(iteE-1)的确定位置
}

dvecIte rndPartition(dvecIte iteB, dvecIte iteE)
{
	srand((unsigned)time(NULL));
	dvecIte iteRnd = iteB + rand() % (iteE-iteB);
	if(iteRnd != iteE-1) {
		double temp = *iteRnd;
		*iteRnd = *(iteE - 1);
		*(iteE-1) = temp;
	}
	return partition(iteB, iteE);
}

double rndSelect(const dvecIte &iteB, const dvecIte &iteE, const size_t i)
{//返回第i小的元素
	if(1 < iteE-iteB) {
		dvecIte ite = rndPartition(iteB, iteE);
		size_t k = ite - iteB + 1;
		if(i == k)
			return *ite;
		else {
			if(i < k)
				return rndSelect(iteB, ite, i);
			else return rndSelect(ite+1, iteE, i-k);
		}
	}
	return *iteB;
}

 int main()
{
	size_t i;//必须输入正整数，不然小数部分会传递给下面的容器
	cout<<"求第几小的元素?"<<endl;
	cin>>i;
	double inData;
	vector<double>dvec;
	cout<<"输入一系列数，以文件结束符结尾!"<<endl;
	while(cin>>inData)
		dvec.push_back(inData);
	cout<<"第"<<i<<"小的元素是:"
		<<setw(5)<<rndSelect(dvec.begin(), dvec.end(), i)<<endl;
	system("PAUSE");
	return EXIT_SUCCESS;
}