002快速排序-第k小数

题目

给定一个长度为 n 的整数数列，以及一个整数 k，请用快速选择算法求出数列从小到大排序后的第 k 个数。

输入格式
第一行包含两个整数 n 和 k。

第二行包含 n 个整数（所有整数均在 1∼109 范围内），表示整数数列。

输出格式
输出一个整数，表示数列的第 k 小数。

数据范围
1≤n≤100000,
1≤k≤n
输入样例：
5 3
2 4 1 5 3
输出样例：
3

#include<iostream>
using namespace std;

const int N = 100010;
int a[N];


int quick_sort(int a[], int l, int r, int k)
{
   if(l == r){
       return a[l];
   }
   int i = l-1;
   int j = r+1;
   int x = a[l+r>>1];
   
   while(i < j){
       do i++; while(a[i] < x);
       do j--; while(a[j] > x);
       if(i< j){
           swap(a[i], a[j]);
       }
   }
   //这里需要考虑相对的j-l+1 
   if(j-l+1>= k){
       return quick_sort(a, l, j, k);
   }else{
       return quick_sort(a, j+1, r, k-(j-l+1));
   }
}

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