算法：快速选择排序

适用场景

要在一个数组中快速找到第k个最大或第k个最小的数。

思路

先利用快速排序将数组分成两个部分，第一部分的数一定小于第二部分的数再判断第一部分数的个数，如果大于k则第k个最小的数一定位于第一部分，如果小于k则这个数在第二部分。利用递归就可以找到这个数，算法复杂度为O（n）。

代码

#include<iostream>
#include<cstdio>
using namespace std;
int n, k;
int a[100010];
int quick_sort(int l, int r, int k)
{
    if(l == r) return a[l];
    int i = l - 1, j = r + 1;
    int x = a[l];
    while(i < j){
        while(a[++i] < x);
        while(a[--j] > x);
        if(i < j) swap(a[i],a[j]);
    }
    int sl = j - l + 1;
    if(k <= sl) return quick_sort(l, j, k);
    if(k > sl) return quick_sort(j + 1, r, k - sl);
}
int main()
{
    cin >> n >> k;
    for(int i = 0; i < n; ++i){
        cin >> a[i];
    }
    cout << quick_sort(0,n-1,k);
    return 0;
}