快速排序题一道

原题链接：acwing786.第k个数

一道快速排序的模板题，关于快速排序的原理和模板，可以参考[模板]快速排序

这里直接贴代码吧，和模板没什么区别，就输出排序后的第k个数而已。

#include<iostream>
using namespace std;
const int N = 1e6 + 10;
int q[N];

void quick_sort(int q[], int l, int r) {
    if(l >= r) {
        return ;
    }
    int x = q[l + r >> 1], i = l - 1, j = 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]);
        }
    }
    quick_sort(q, l, j);
    quick_sort(q, j + 1, r);
}

int main() {
    int n, k;
    scanf("%d%d", &n, &k);
    for(int i = 0; i < n; ++i) {
        scanf("%d", &q[i]);
    }
    quick_sort(q, 0, n - 1);
    printf("%d\n", q[k - 1]);
    return 0;
}