Cod sursa(job #2065917)

Utilizator Horia14Horia Banciu Horia14 Data 14 noiembrie 2017 15:07:45
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.95 kb
#include<cstdio>
#include<cstdlib>
#include<ctime>
#define MAX_N 3000000
using namespace std;

int v[MAX_N], n;

int main() {
    int k, i, Begin, End, b, e, pivot, aux;
    FILE *fin, *fout;
    fin = fopen("sdo.in","r");
    fout = fopen("sdo.out","w");
    fscanf(fin,"%d%d",&n,&k);
    for(i=0; i<n; i++)
        fscanf(fin,"%d",&v[i]);
    srand(time(NULL));
    k--;
    Begin = 0, End = n - 1;
    while(Begin < End) {
        b = Begin;
        e = End;
        pivot = v[Begin + rand()%(End-Begin+1)];
        while(b <= e) {
            while(v[b] < pivot) b++;
            while(v[e] > pivot) e--;
            if(b <= e) {
                aux = v[b];
                v[b] = v[e];
                v[e] = aux;
                b++,e--;
            }
        }
        if(k <= e)
            End = e;
        else if(k >= b)
            Begin = b;
        else Begin = End = k;
    }
    fprintf(fout,"%d\n",v[k]);
    fclose(fin);
    fclose(fout);
    return 0;
}