Cod sursa(job #2014233)

Utilizator smatei16Matei Staicu smatei16 Data 23 august 2017 11:38:58
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.7 kb
#include <cstdio>
#include <algorithm>
#include <ctime>
using namespace std;
int a[3000003],poz,n,k,i;
int main()
{
    freopen("sdo.in","r",stdin);
    freopen("sdo.out","w",stdout);
    scanf("%d %d",&n,&k);
    srand(time(0));
    int st=1;int dr=n;
    for(i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    while(st<=dr){
        int ha=rand()%(dr-st+1)+st;
        int pivot=a[ha];
        poz=st;
        swap(a[ha],a[dr]);
        for(i=st;i<=dr;i++)
            if(a[i]<=pivot)swap(a[i],a[poz++]);
        poz--;
        if(poz<k)st=poz+1;
        else if(poz>k)dr=poz-1;
        else {
            printf("%d",pivot);
            break;
        }
    }


    return 0;
}