Cod sursa(job #1841252)

Utilizator AnduuFMI Alexandru Banu Anduu Data 5 ianuarie 2017 14:32:13
Problema Statistici de ordine Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include <fstream>
#include <vector>
#include <iostream>
using namespace std;

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

    while (i <= j) {
        while (p[i] < pivot) {
            ++i;
        }
        while (p[j] > pivot) {
            --j;
        }
        if (i <= j) {
            if (i == mid) {
                mid = j;
            } else if (j == mid) {
                mid = i;
            }
            swap(p[i++], p[j--]);
        }
    }
    for (int q = 0; q < 8; ++q) {
        cout << p[q] << ' ';
    }
    cout << '\n';
    if (lo < j || i < hi) {
    if (mid < k) {
        qsort (p, mid + 1, hi, k);
    } else
    if (k <= mid) {
        qsort (p, lo, mid, k);
    }
    }
}

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;
}