Cod sursa(job #1731307)

Utilizator Lungu007Lungu Ionut Lungu007 Data 18 iulie 2016 17:41:02
Problema Statistici de ordine Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.96 kb
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <ctime>
using namespace std;
#define NMAX 3000001
ifstream in("sdo.in");
ofstream out("sdo.out");

int i,n,a[NMAX];

int part(int st,int dr)
{
    int z = rand()%(dr-st+1) + st;
    cout << a[z] << " ";
    swap(a[z],a[dr]);

    int x = a[dr];
    int i = st-1;
    for(int j = st;j<dr;j++)
    {
        if(a[j]<=x)
        {
            i++;
            swap(a[j],a[i]);
        }
    }
    swap(a[dr],a[i+1]);
    return i+1;
}

int stat(int st,int dr)
{
    if(st==dr) return a[st];

    int mij = part(st,dr),k = mij-st+1;

    if(k == i) return a[mij];

    if(k>i)
    {
        stat(st,mij-1);
    }
    else
    {
        stat(mij+1,dr);
    }
}

int main()
{
    srand(time(0));
    in >> n >> i;
    for(int i=0;i<n;i++) in >> a[i];

    out << stat(0,n-1);
//    for(int i=0;i<n;i++)
//    {
//        out << a[i] << " ";
//    }
    return 0;
}