Pagini recente » Istoria paginii utilizator/georgianapurcaru | Istoria paginii runda/sda | Cod sursa (job #2155678) | Istoria paginii runda/o_o/clasament | Cod sursa (job #1992028)
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <ctime>
using namespace std;
ifstream in("sdo.in");
ofstream out("sdo.out");
#define ll long long
#define pb push_back
const int inf = 1e9 + 5;
const int NMax = 3e6 + 5;
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];
in.close();out.close();
return 0;
}
void solve(int st,int dr) {
if (st >= dr) {
return;
}
int idx = part(st,dr);
if (K <= idx) {
solve(st,idx);
}
if (K > idx) {
solve(idx+1,dr);
}
}
int part(int st,int dr) {
int idx = rand()%(dr-st+1) + st;
//swap(v[idx],v[1]);
int pivot = v[idx];
int i=st-1,j = dr+1;
while (true) {
do {
++i;
}
while (v[i] < pivot);
do {
--j;
}
while (pivot < v[j]);
if (i < j) {
swap(v[i],v[j]);
}
else {
return j;
}
}
}