快排(第k个数)

第k个数

如题：

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

输入格式
第一行包含两个整数 n 和 k。
第二行包含 n 个整数（所有整数均在 1∼109 范围内），表示整数数列。
输出格式
输出一个整数，表示数列的第 k 小数。
数据范围
1≤n≤100000,
1≤k≤n

代码如下

#include<iostream>
using namespace std;
const int N=1e5+5;
int a[N],n,k;
void quick_sort(int a[],int l,int r)
{
    int i=l,j=r,mid=a[l+r>>1];
    while(i<j)
    {
        while(a[i]<mid)i++;
        while(a[j]>mid)j--;
        if(i<=j)
        {
            swap(a[i],a[j]);
            i++,j--;
        }
    }
    if(l<j&&k<=j)quick_sort(a,l,j);
    if(i<r&&k>=i)quick_sort(a,i,r);
}
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)cin>>a[i];
    quick_sort(a,1,n);
    cout<<a[k]<<endl;
    return 0;
}

注释：

1.在递归时加上判断条件k<=j和k>=i可减少不必要的递归
2.在if(i<=j)中i<=j目的是确定下一次的递归边界，防止无穷递归