Cod sursa(job #3274594)

Utilizator andreipirjol5Andrei Pirjol andreipirjol5 Data 7 februarie 2025 14:03:24
Problema Statistici de ordine Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.01 kb
#include <iostream>
#include <vector>
#include <cstdlib>

using namespace std;

int partition(vector<int>& arr, int left, int right) {
    int pivot = arr[right];
    int i = left;
    for (int j = left; j < right; j++) {
        if (arr[j] <= pivot) {
            swap(arr[i], arr[j]);
            i++;
        }
    }
    swap(arr[i], arr[right]);
    return i;
}

int quick_select(vector<int>& arr, int left, int right, int k) {
    if (left == right)
        return arr[left];

    int pivot_index = partition(arr, left, right);
    int order = pivot_index - left + 1;

    if (order == k)
        return arr[pivot_index];
    else if (order > k)
        return quick_select(arr, left, pivot_index - 1, k);
    else
        return quick_select(arr, pivot_index + 1, right, k - order);
}

int main() {
    int n, k;
    cin >> n >> k;
    vector<int> arr(n);
    for (int i = 0; i < n; i++)
        cin >> arr[i];

    cout << quick_select(arr, 0, n - 1, k) << endl;
    return 0;
}