Cod sursa(job #1017889)

Utilizator stefyvoltFMI Stefan Niculae stefyvolt Data 28 octombrie 2013 16:39:54
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 kb
#include <iostream>
#include <fstream>
#include <ctime>
#include <cstdlib>

using namespace std;

ifstream f ("sdo.in");
ofstream g ("sdo.out");
int n, k, v[3000001];

void stat_de_ord(int a=1, int b=n)      // quicksort modificat
{
    int i=a, j=b, p=v[rand()%(b-a+1) + a];     // v[i=a ... p ... j = b]

    while(i <= j)                       // cat timp i nu a ajuns la j
    {
        while(v[i] < p) i++;            // crestem i pana ajungem la pivot
        while(v[j] > p) j--;            // scadem j pana la pivot

        if(i <= j)                      // daca elementele la care ne-am oprit sunt de parti opuse ale pivotului
        {
            int aux = v[i];
        v[i] = v[j];            // le interschimbam valorile
            v[j] = aux;
        i++; j--;                   // si le lasam sa continue drumul
        }
    }

    if      (k <= j) stat_de_ord(a,j);        // apelam QS in subvectorul din stanga
    else if (k >= i)  stat_de_ord(i,b);        // apelam QS in dreapta
}

int main()
{
    f >> n >> k;
    for(int i=1; i<=n; i++) f >> v[i];

    stat_de_ord();

    g << v[k];

    return 0;
}