题目描述

条件是k<n,求第k个小的数

时间复杂度 $O(n)$

C++ 代码

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