线性求第k大

快排变种.

快排每次只进行部分排序,进入左边或者右边或者当前mid就是答案.

据说期望值是O(n)

然后STL中的 nth_element也是用这个思想.

#include <cstdio>
// #include <windows.h>

using namespace std;

const int maxn = 1e6+;

int arr[maxn];
int n, k;

int _sort(int l, int r) {

    int mid = (l + r) / ;
    int ll = l, rr = r;
    int val;
    val = arr[mid];
    arr[mid] = arr[l];
    while (l < r) {
        while (arr[r] > val && l < r) r--;
        arr[l] = arr[r];
        while (arr[l] <= val && l < r) l++;
        arr[r] = arr[l];
    }
    arr[l] = val;
    if (l == k) return arr[l];
    if (l -  > ll && ll<=k && k<=l-) return _sort(ll, l-);
    else if (rr > l +  && l+<=k && k>=rr) return _sort(l+, rr);
    else if (l- == k) return arr[l-];
    else if (l+ == k) return arr[l+];
    else return -;
}

int main() {
    //int n;
    scanf("%d%d", &n, &k);
    for (int i=; i<n; ++i) scanf("%d", &arr[i]);
    k = n - k;
    printf("%d \n", _sort(, n-));
//    for(int i=0; i<n; ++i) printf("%d ", arr[i]);
//    system("pause");

    return ;
}