Cod sursa(job #2126290)

Utilizator MaligMamaliga cu smantana Malig Data 9 februarie 2018 14:45:22
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include <bits/stdc++.h>

#if 1
#define pv(x) cout<<#x<<" = "<<x<<"; ";cout.flush()
#define pn cout<<endl
#else
#define pv(x)
#define pn
#endif

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

using ll = long long;
using ull = unsigned long long;
using uint = unsigned int;
#define pb push_back
#define mp make_pair
const int NMax = 3e6 + 5;
const int inf = 1e9 + 5;
using zint = int;

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(NULL));
    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-1);
    }
    else if (idx < K) {
        solve(idx+1,dr);
    }
}

int part(int st,int dr) {
    int idx = rand() % (dr-st+1) + st;
    swap(v[dr],v[idx]);

    int i = st-1;
    for (int j=st;j <= dr;++j) {
        if (v[j] <= v[dr]) {
            swap(v[++i], v[j]);
        }
    }

    return i;
}