Cod sursa(job #1779903)

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

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