类似快排的方法解决无序数组中第k小数。
效率和快排一样吧,Onlogn ~ On^2.不过比(快排后输出)要快,因为每次对于k不在的区间就直接舍弃不用处理了。
复习快排。
利用了快排的思想,每次随机选择一个中间点,这里是选择区间最左边的数piv作为中间点,然后将数组中比piv小的依次放到左边(从最左往右),比piv大的依次放到右边(从最右往左),留下中间的位置mid给piv,如果mid == k,那么piv就是k小数,否则选择k所在区间,继续重复上一步操作。
const int N = 1111111;
int a[N];
int main(){
int n,k,le,ri,m;
scanf("%d%d",&n,&k);
for(int i=0;i<n;i++){
a[i] = in();
}
if(n <k) puts("-1");
else{
int l = 0,r = n-1;
while(l <= r){
m = a[le = l] , ri = r;
while(le < ri){ //手动模拟一下应该很好理解
while(le < ri && a[ri] > m) --ri;
a[le] = a[ri];
while(le < ri && a[le] < m) ++le;
a[ri] = a[le];
}
a[le] = m;
if(le < k-1) l = le + 1;
else r = le - 1;
}
printf("%d\n",a[r + 1]);
}
return 0;
}