Cod sursa(job #1779813)

Utilizator dragos_vecerdeaVecerdea Dragos dragos_vecerdea Data 15 octombrie 2016 17:00:49
Problema Statistici de ordine Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#include <stdio.h>
#include <cstdlib>
using namespace std;
FILE *fin = fopen("sdo.in" , "r");
FILE *fout = fopen("sdo.out" , "w");
#define MAX 3000000
int v[MAX+1];
int n , k;
void schimb(int i , int j)
{
    int aux=v[i];
    v[i]=v[j];
    v[j]=aux;
}
int pivot(int st , int dr , int p)
{
    int poz=1;
    for(int i=1;i<=n;i++)
    {
        if(v[p]>v[i]) poz++;
    }
    schimb(poz , p);
    int i1 , i2;
    while(i1<poz && i2<=n)
    {
        if(v[i1]>v[poz] && v[i2]<v[poz])
        {
            schimb(i2 , i1);
            i2++;
            i1++;
        }
        else
        {
            if(v[i1]>v[poz] && v[i2]>v[poz]) i2++;
            else if(v[i1]<v[poz] && v[i2]<v[poz])i1++;
        }
    }
    return poz;
}
void qs(int st , int dr)
{
    if(st<dr)
    {
        int p=st+(dr-st)/2;
        p=pivot(st , dr , p);
        if(k<p) qs(st , p-1);
        else
        {
            if(k==p) return;
            else qs(p+1,dr);
        }

    }
    else return;
}
int main()
{
    fscanf(fin , "%d%d" , &n , &k);
    for(int i=1;i<=n;i++)
    {
        fscanf(fin , "%d" , &v[i]);
    }
    qs(1 , n);
     fprintf(fout , "%d" , v[k]);
    return 0;
}