Pagini recente » Cod sursa (job #604264) | Cod sursa (job #333388) | Cod sursa (job #741500) | Cod sursa (job #3038010) | Cod sursa (job #3274596)
#include <iostream>
#include <cstdlib>
using namespace std;
int partition(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(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;
int arr[n];
for (int i = 0; i < n; i++)
cin >> arr[i];
cout << quick_select(arr, 0, n - 1, k) << endl;
return 0;
}