acwing 786.第k个数

也是快速排序的做法，满足前面几个数小于mid，小于mid的总数有没有大于k，如果大于，在左边区间继续快速查找，如果小于，则在右边区间继续查找

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

输入格式

第一行包含两个整数 n� 和 k�。

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

输出格式

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

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

const int maxn=100010;
int n,k,q[maxn];

int quick_select(int l,int r,int k)
{   
    if(l==r)    return q[l];
    
    int mid=q[(l+r)>>1],i=l-1,j=r+1;
    while(i<j)
    {
        do i++;while(q[i]<mid);
        do j--;while(q[j]>mid);
        if(i<j) swap(q[i],q[j]);
    }
    
    int s1=j-l+1;
    if(k<=s1)   return quick_select(l,j,k);
    else    return quick_select(j+1,r,k-s1);
}

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