Pagini recente » Cod sursa (job #2446653) | Cod sursa (job #1008737) | Cod sursa (job #1766619) | Cod sursa (job #1049998) | Cod sursa (job #1982421)
#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;
}
}
}