洛谷1923 排序

洛谷1923

错误

这道题用的快排，但是非常卡时间，最后将快排优化，

新学

stl中 nth_element(数组名,数组名+第k小元素,数组名+元素个数)；将第k小元素找出，最后直接输出就行

//判断k所在的某一段位置
if(k<=y)
    qsort1(l,y);
    else if(x<=k)
    qsort1(x,r);
    else{
        cout<<arr[k];
        exit(0);
    }

代码

#include<iostream>
#include<cstdio>
using namespace std;
int k;
int arr[50000005];
void qsort1(int l,int r){
    int mid = arr[(l+r)/2];
    int x=l,y=r;
    do{
        while(arr[x]<mid)x++;
        while(arr[y]>mid)y--;
        if(x<=y){
            swap(arr[x],arr[y]);
            x++;y--;
        }
    }while(x<=y);

    if(k<=y)
    qsort1(l,y);
    else if(x<=k)
    qsort1(x,r);
    else{
        cout<<arr[k];
        exit(0);
    }
}
int main(){
    int n;
    scanf("%d%d",&n,&k);
   
    for(int i = 0 ;i<n;i++){
       scanf("%d",&arr[i]);
    }
    qsort1(0,n-1);
}