蓝桥杯day6-快排：acwing-786. 第k个数

 

#include <iostream>

using namespace std;

const int N = 100010;

int n, k;

int q[N];

int quick_sort(int l, int r, int k)//此处由于函数有返回值，故不可以用void
{
    if (l >= r) return q[l];
    //递归时，一定保证第k小数，一定是在区间l~r里面的
    //当区间长度为1时，并且答案位于其中，该数就一定是答案 返回q[l]或q[r]

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

    if (k<=j) return quick_sort(l, j, k);//如果第 k 大的数在左半边，则递归左半边寻找
    else return quick_sort( j + 1, r, k);// 否则递归右半边
}

int main()
{
    scanf("%d%d", &n, &k);

    for (int i = 0; i < n; i ++ ) scanf("%d", &q[i]);

    // 返回的就是第 k 大的数，注意这里传入的是下标 k - 1
    cout << quick_sort( 0, n - 1, k-1) << endl;

    return 0;
}