acwing 786第K个数

有两种解法，首先用快速排序算法，这种算法时间复杂度是o(nlogn)

#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int q[N];
void quick_sort(int a[],int low,int high){
    if(low >=high) return;
    int i = low - 1, j = high + 1, x = a[(low + high)>>1];

    while(i < j){
        do i ++;while(a[i] < x);

        do j --;while(a[j] > x);
        if (i < j) swap(a[i],a[j]);
    }
    quick_sort(a,low,j);
    quick_sort(a,j + 1,high);

}
int main(){
   int n,k;
    cin>>n>>k;
    for(int i = 0; i< n; i ++)
    cin>>q[i];
    quick_sort(q,0,n - 1);
    cout<<q[k - 1];
}

第二种算法是快速选择算法，其思想和快速排序思想类似，只不过不用递归排序左右两边，只需递归单侧即可，因为是求第k个数，所以当一轮排序之后，左边所有的数字都比基准元素要小，右边所有数字都比基准元素要大，如果k的值小于左侧元素个数，说明第k个数一定在左侧，那么就递归左侧就行，不用再递归右侧，如果k的值大于左侧元素个数，说明第k个数一定在右侧，那么就递归右侧就行，不用再递归左侧了，当递归过程中，整个区间只剩一个元素的时候就是我们要找的元素,所以遍历的时间复杂度是o（n + n/2 + n/4 + …）=o(2n)，所以时间复杂度就是o(n)

#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int q[N];
int quick_select(int low,int high,int k){//每次递归，第k个数一定在low和high之中
    if (low >= high) return q[low];//所以当low和high之间只剩一个元素的时候，这个元素就是第k个数
    int i = low - 1,j = high + 1,x = q[low];
    while(i < j){
        do i ++; while(q[i] < x);
        do j --; while(q[j] > x);
        if(i < j) swap(q[i],q[j]);
    }
    int left_length = j - low + 1;
    if(k <= left_length)   quick_select(low, j, k);
    else   quick_select(j + 1, high, k - left_length);
}
int main(){
   int n,k;
    cin>>n>>k;
    for(int i = 0; i< n; i ++)
    cin>>q[i];
    cout<<quick_select(0,n - 1, k);

}