??采用的算法思想和快速排序十分相似,选取基准进行一次快排,得到一个基准和左侧小于基准右侧大于基准的数组。然后判断基准的位置,如果第k个数在左边便接着向左递归,如果在右侧便接着向右侧递归。
#include<iostream>
#include<set>
#include<cmath>
#include<vector>
#include<map>
#include<string>
#include<sstream>
using namespace std;
//进行分割排序,并且返回排序后基准的位置
template <typename Type>
int Partition(Type a[],int p,int r){
int i=p,j=r+1; //这里的j=r+1,之所以奖赏一是因为后面的--j,要先做减法,为了防止最后一个元素称为漏网之鱼
Type x=a[p]; //这里将数组的起始点下标p指向的元素作为基准
//将小于x的元素交换到左边区域,将大于x的元素交换到右边区域
while(true){
while(a[++i]<x&&i<r); //先扫描左边的,直到找到比x大的元素停下来
while(a[--j]>x); //找到右边比x小的元素停下来
if(i>=j)break; //结束条件
Swap(a[i],a[j]) //将他俩进行交换
}
//这里j指向的要小于等于x
a[p]=a[j];
a[j]=x;
return j;
}
//进行随机分割,避免数组整体倒序造成时间复杂度升高
template <typename Type>
int RandomizedPartition(Type a[],int p,int r){
int i=random(p, r);
swap(a[i],a[p]);
return Partition(a,p,r);
}
/*
线性时间选择第k个数:
a[] :数组
p :数组起点
r :数组终点
k :第k个数
*/
template <typename Type>
Type RandomizedSelect(Type a[],int p,int r,int k){
//如果找到k,即分割到只有一个数的时候
if(p==r) return a[p];
int i=RandomizedPartition(a,p,r);
int j=i-p+1; //用来判断第k个数是在基准的左边还是右边
if(k<=j)return RandomizedSelect(a,p,i,k); //如果在左边
else return RandomizedSelect(a,i+1,r,k-j); //如果在右边
}
原文:https://www.cnblogs.com/3236676588buladuo/p/14723910.html