Cod sursa(job #1265961)

Utilizator bogdan10bosBogdan Sitaru bogdan10bos Data 17 noiembrie 2014 23:33:13
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.12 kb
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <ctime>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <deque>

#define INF (1<<30)
#define mod 666013

using namespace std;
int n, k, i, a[3000005];

void chg(int x, int y)
{
    int aux=a[x];
    a[x]=a[y];
    a[y]=aux;
}

int piv(int st, int dr)
{
    int x=st-1, y=dr+1;
    int rnd=rand()%(dr-st+1);
    int pv=a[st+rnd];

    while(1)
    {
        x++;
        while(a[x]<pv)
            x++;

        y--;
        while(a[y]>pv)
            y--;

        if(x<y)
            chg(x, y);
        else
            return y;
    }

    return 0;
}

void sdo(int st, int dr, int k)
{
    if(st>=dr) return;
    int w=piv(st, dr);
    int q=w-st+1;

    if(q>=k)
        sdo(st, w, k);
    else
        sdo(w+1, dr, k-q);
}

int main()
{
    freopen("sdo.in", "r", stdin);
    freopen("sdo.out", "w", stdout);
    srand(time(0));

    scanf("%d%d", &n, &k);
    for(i=1;i<=n;i++)
        scanf("%d", &a[i]);
    sdo(1, n, k);
    printf("%d", a[k]);
    return 0;
}