Cod sursa(job #1477700)

Utilizator AlexandruValeanuAlexandru Valeanu AlexandruValeanu Data 26 august 2015 19:49:07
Problema Statistici de ordine Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.8 kb
#include <bits/stdc++.h>

using namespace std;

void kth_element(int A[], int l, int r, int k)
{
    if (l >= r)
        return;

    int i = l, j = r;
    int p = A[l + rand() % (r - l + 1)];

    do
    {
        while (A[i] < p) i++;
        while (A[j] > p) j--;

        if (i <= j)
        {
            swap(A[i], A[j]);

            i++;
            j--;
        }

    } while (i < j);

    if (i <= k && k <= r)
        kth_element(A, i, r, k);

    if (l <= k && k <= j)
        kth_element(A, l, j, k);
}

const int MAX_N = 3000000 + 1;

int A[MAX_N];
int N, K;

int main()
{
    srand(time(nullptr));

    ifstream in("sdo.in");
    ofstream out("sdo.out");

    in >> N >> K;

    for (int i = 0; i < N; ++i)
        in >> A[i];

    kth_element(A, 0, N - 1, K - 1);

    out << A[K - 1];

    return 0;
}