Cod sursa(job #1982421)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 18 mai 2017 19:20:31
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.85 kb
#include <iostream>
#include <fstream>
#include <math.h>
#include <algorithm>
#include <cstdlib>
#include <time.h>

using namespace std;
ifstream in("sdo.in");
ofstream out("sdo.out");

#define ll long long
#define pb push_back
const int NMax = 5e6 + 5;

// se gaseste a K-a statistica de ordine
// folosind o varianta modificata a algoritmului quick sort

int N,K;
int v[NMax];

void solve(int,int);
int part(int,int);

int main() {
    in>>N>>K;
    for (int i=1;i <= N;++i) {
        in>>v[i];
    }

    srand(time(0));
    solve(1,N);

    out<<v[K]<<'\n';

    in.close();out.close();
    return 0;
}

// functia solve va sorta doar anumite parti ale lui v,
// suficient cat sa se puna al K-lea minim pe pozitia corecta
// si sa se poata astfel identifica acesta accesand v[K]
void solve(int st,int dr) {
    if (st >= dr) {
        return;
    }

    int idx = part(st,dr);
    // idx reprezinta pozitia corecta a pivotului utilizat in subrutina part
    // daca acesta este = cu K, s-a gasit a K-a statistica de ordine si functia se termina
    // daca idx > K, se realizeaza o sortare partiala in partea din stanga
    // daca idx < K, se realizeaza o sortare partiala in partea din dreapta
    if (K < idx) {
        solve(st,idx-1);
    }
    if (K > idx) {
        solve (idx+1,dr);
    }
}

// la fel ca la algoritmul quick sort
// (schema de partitie Hoare)
int part(int st,int dr) {
    int idx = rand() % (dr-st+1) + st;
    swap(v[st],v[idx]);

    int i = st-1,
        j = dr+1;
    while (true) {
        do {
            ++i;
        }
        while (v[i] < v[st]);

        do {
            --j;
        }
        while (v[j] > v[st]);

        if (i < j) {
            swap(v[i],v[j]);
        }
        else {
            swap(v[st],v[j]);
            return j;
        }
    }
}