Pagini recente » Cod sursa (job #1287013) | Cod sursa (job #3228392) | Cod sursa (job #1342122) | Cod sursa (job #2211132) | Cod sursa (job #3274594)
#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;
}