Cod sursa(job #2277614)

Utilizator badea_adi1999Badea Adrian Catalin badea_adi1999 Data 6 noiembrie 2018 16:59:49
Problema Statistici de ordine Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <iostream>
#include <cstdio>
#include <ctime>
#include <cstdlib>
#include <algorithm>
#define nmax 3000005
using namespace std;
int n,k,v[nmax];

int part(int a[],int st,int dr)
{
    int i= st , j= dr ;
    int p= st+ rand()%(dr- st + 1);
    swap(a[st],a[p]);

    while(1) {
        if (a[i]>a[j]) {
            swap(a[j],a[i+1]);
            swap(a[i],a[i+1]);
            i++;
        }
        else
            j--;
        if (i>=j)
            break;
    }
    return i;
}
int sel(int a[], int st, int dr, int k)
{
    if (st == dr)
        return  a[st];
    int q= part(a, st, dr);
    int t= q - st +1;
    if (t >= k)
        sel(a, st, q, k);
    else
        sel(a, q+1, dr, k-t);
}
int main()
{
    freopen("sdo.in","r",stdin);
    freopen("sdo.out","w",stdout);
    srand(time(NULL));
    scanf("%d %d", &n, &k);
    for (int i = 1; i<=n; i++)
        scanf("%d", &v[i]);

    printf("%d",sel(v,1,n,k));

    return 0;
}