Cod sursa(job #2281345)

Utilizator pasoi_stefanPasoi Stefan pasoi_stefan Data 12 noiembrie 2018 08:24:00
Problema Statistici de ordine Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.62 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,j=d;
    while(i<=j){
        while(a[i]<pivot) ++i;
        while(a[j]>pivot) --j;
        if(i<=j){
            swap(a[i],a[j]);
            ++i; --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];
}