Cod sursa(job #855030)

Utilizator UnforgivenMihai Catalin Botezatu Unforgiven Data 14 ianuarie 2013 16:17:04
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.05 kb
#include <fstream>
#include <cstdlib>
#include <ctime>
#define NMAX 3000001

using namespace std;

int v[NMAX];
int N;
int K;

int partition(int left , int right)
{
    int i=left - 1,j=right + 1,aux;
    int p = v[left + (rand() % (right - left + 1))];
    while(true)
    {
        do
        {
            i++;
        }while(v[i]<p);
        do
        {
            j--;
        }while(v[j]>p);
        if(i< j)
        {
            aux=v[i];
            v[i]=v[j];
            v[j]=aux;
        }
        else
        {
            return j;
        }
    }
}

void SDO(int left , int right , int K)
{
    while (left != right)
    {

        int q = partition(left , right);
        int t = q - left + 1;
        if (t >= K) right = q;
        else left = q + 1 , K = K - t ;
    }
}

int main()
{
    srand(time(NULL));
    ifstream input("sdo.in");
    ofstream output("sdo.out");
    input >> N >> K;
    for (int i = 1 ; i<=N;i++)
    {
        input >> v[i];
    }
    SDO(1,N,K);
    output << v[K];
    input.close();
    output.close();
    return 0;
}