Cod sursa(job #2909830)

Utilizator Mihai145Oprea Mihai Adrian Mihai145 Data 16 iunie 2022 09:18:39
Problema Statistici de ordine Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.72 kb
#include <fstream>
#include <vector>

std::ifstream cin("sdo.in");
std::ofstream cout("sdo.out");

int select(std::vector<int> &v, int stat) {
    if(v.size() == 1) {
        return v[0];
    }

    int piv = v[rand() % v.size()];
    std::vector<int> left, right;
    for(int x : v) {
        if(x < piv) { left.push_back(x); }
        else { right.push_back(x); }
    }
    v.clear();

    if(left.size() == stat) {
        return piv;
    }

    if(left.size() > stat) {
        return select(left, stat);
    }

    return select(right, stat - left.size());
}

int main() {
    int n, k; cin >> n >> k;

    std::vector<int> v(n);
    for(int i = 0; i < n; i++) {
        cin >> v[i];
    }

    cout << select(v, k - 1);
    return 0;
}