/*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;
}
算法导论CLRS 9.2 选择第i小的元素
最新推荐文章于 2020-04-20 18:02:10 发布