Cod sursa(job #2281342)

Utilizator pasoi_stefanPasoi Stefan pasoi_stefan Data 12 noiembrie 2018 08:18:57
Problema Statistici de ordine Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.61 kb
#include<fstream>
#include<ctime>
#include<cstdlib>
using namespace std;
ifstream cin("sdo.in");
ofstream cout("sdo.out");
int n,k,a[3000005];
void Kth_element(int s,int d){
    if(s>=d) return;
    int pivot=a[s+rand()%(d-s+1)];
    int i=s-1,j=d+1;
    while(i<j){
        do{++i;}while(a[i]<pivot);
        do{--j;}while(a[j]>pivot);
        if(i<j){
            swap(a[i],a[j]);
        }
    }
    if(k<=j) Kth_element(s,j);
    else Kth_element(j+1,d);
}
int main(){
    srand(time(NULL));
    cin>>n>>k;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    Kth_element(1,n);
    cout<<a[k];
}