第K个数 —快速选择(快排)

 输入样例：

5 3 
2 4 1 5 3

期望输出：

3

代码实现

　　

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+9;
int n,m;
int a[N];

int quick_sort(int l,int r,int m)
{
    //每次都保证第k个数在区间里面
    if(l==r) return a[l];

    int i=l,j=r,x=a[l];

    while(i<j)
    {
        while(i<j&&a[j]>x)
            j--;
        a[i]=a[j];
        while(i<j&&a[i]<x)
            i++;
        a[j]=a[i];
    }
    a[j] =x;

    //左半边的元素数量
    int numl = j-l+1;
    //每次都保证第k个数在区间里面
    if(numl<m) return quick_sort(j+1,r,m-numl);
    else return quick_sort(l,i,m);
}

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

    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
     cout<<quick_sort(0,n-1,m);
}