Pagini recente » Cod sursa (job #2410590) | Cod sursa (job #540236) | Cod sursa (job #1800454) | Cod sursa (job #1969100) | Cod sursa (job #3233222)
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
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 quickselect(vector<int>& arr, int left, int right, int k) {
if (left == right) {
return arr[left];
}
int pivotIndex = partition(arr, left, right);
if (k == pivotIndex) {
return arr[k];
} else if (k < pivotIndex) {
return quickselect(arr, left, pivotIndex - 1, k);
} else {
return quickselect(arr, pivotIndex + 1, right, k);
}
}
int main() {
ifstream infile("sdo.in");
ofstream outfile("sdo.out");
if (!infile || !outfile) {
cerr << "Error opening file" << endl;
return 1;
}
int n, k;
infile >> n >> k;
vector<int> arr(n);
for (int i = 0; i < n; ++i) {
infile >> arr[i];
}
int result = quickselect(arr, 0, n - 1, k - 1);
outfile << result << endl;
infile.close();
outfile.close();
return 0;
}