基础算法：快速排序

题目描述

参考代码

#include<iostream>
using namespace std;
/*
快速排序：O(nlogn)
1、找到分界点x，q[l]、q[r]、q[(l+r)/2]
2、左边所有数小于等于x，右边所有数大于等于x
3、递归排序左边区间，递归排序右边区间
*/
/*
快速选择：O(n)
1、找到分界点x，q[l]、q[r]、q[(l+r)/2]
2、左边所有数小于等于x，右边所有数大于等于x
3、根据k的大小选择递归区间
*/

const int N = 100010;

int n, k;
int q[N];

int quick_sort(int l, int r, int k)
{   
    // 递归退出条件
    if (l >= r) return q[l];

    int x = q[l], i = l - 1, j = r + 1;
    while (i < j)
    {
        while (q[++ i] < x);
        while (q[-- j] > x);
        if (i < j) swap(q[i], q[j]);
    }

    int sl = j - l + 1;          // 左边区间元素个数
    if (k <= sl) return quick_sort(l, j, k);

    return quick_sort(j + 1, r, k - sl);
}

int main()
{
    cin >> n >> k;

    for (int i = 0; i < n; i++) cin >> q[i];

    cout << quick_sort(0, n - 1, k) << endl;

    return 0;
}