Cod sursa(job #1837757)

Utilizator AnduuFMI Alexandru Banu Anduu Data 30 decembrie 2016 13:56:29
Problema Statistici de ordine Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 0.93 kb
#include <fstream>
#include <vector>
using namespace std;

void qsort(vector<int> &p, int lo, int hi, int k) {
    int i = lo, j = hi, pivot = p[(lo + hi) / 2];

    while (i <= j) {
        while (p[i] < pivot) {
            ++i;
        }
        while (p[j] > pivot) {
            --j;
        }
        if (i <= j) {
            swap(p[i++], p[j--]);
        }
    }
    if (lo < j || i < hi) {
    if (j <= k) {
        qsort (p, j, hi, k - j);
    } else
    if (k <= i) {
        qsort (p, lo, i, k);
    } else {
        qsort (p, j, i, k - j);
    }
    }
}

int main() {
    int n, k;
    vector<int> vec;

    ifstream in("sdo.in");
    in >> n >> k;
    for (int i = 0; i < n; ++i) {
        int x;
        in >> x;
        vec.push_back(x);
    }
    in.close();

    qsort(vec, 0, n - 1, k - 1);

    ofstream out("sdo.out");
    out << vec[k - 1] << '\n';
    out.close();

    return 0;
}