Cod sursa(job #1941188)

Utilizator dorin31Geman Dorin Andrei dorin31 Data 27 martie 2017 01:15:20
Problema Statistici de ordine Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.79 kb
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <ctime>

#define maxN 3000003

using namespace std;

ifstream fin("sdo.in");
ofstream fout("sdo.out");

int n,k,v[maxN];

int pivot(int left, int right)
{
    int i=left, j=right, x=v[left];
    while (i<j)
    {
        while (v[i]<x) ++i;
        while (v[j]>=x) --j;
        if (i<j) swap(v[i],v[j]);
    }
    return i;
}

void quickSelect(int left, int right, int k)
{
    if (left==right) return;

    int x=pivot(left,right);
    int p=x-left+1;

    if (p>=k) quickSelect(left,p,k);
    else quickSelect(p+1,right,k-p);
}

int main()
{
    //srand(time(NULL));
    fin>>n>>k;
    for (int i=1; i<=n; ++i)
        fin>>v[i];

    quickSelect(1,n,k);
    fout<<v[k];
    return 0;
}