P1923 【深基9.例4】求第 k 小的数

欸嘿，典型的求解第k小的数的问题

算法中使用了快速排序

关键之处在于原本快排需要两边都进行排序，但现在我们只关心第k小的数，所以，如果在ll比k大那么就排左半边，比k小就比右半边，从而实现时间复杂度的下降

 

#include<bits/stdc++.h>

using namespace std;

const int MAXN=5000005;

int n,k;
int a[2][MAXN];

void read(int &p)
{
    p=0;char s=getchar();
    while(!isdigit(s)) s=getchar();
    for(;isdigit(s);s=getchar()) p=p*10+s-'0';
}
//利用快排 
int sortx(int l,int r,int flag)
{
    int tem=a[flag][(l+r)/2];int ll=l,rr=r;
    //cout<<tem<<" ";
    for(int i=l;i<=r;i++)
    {
        if(i==(l+r)/2) continue;
        if(a[flag][i]<tem) a[flag^1][ll++]=a[flag][i];
        else a[flag^1][rr--]=a[flag][i];
    } 
    a[flag^1][ll]=tem;
    //cout<<ll<<" "<<rr<<endl;
    if(ll==k) return tem; 
    if(ll<k) return sortx(ll+1,r,flag^1);
        else return sortx(l,ll,flag^1);
}


int main()
{
    cin>>n>>k;k++;
    for(int i=1;i<=n;i++) read(a[0][i]);
    //for(int i=1;i<=n;i++) cout<<a[0][i]<<" ";
    cout<<sortx(1,n,0);
    
    
    
    return 0;
}