Pagini recente » Cod sursa (job #3003432) | Cod sursa (job #2189465) | Cod sursa (job #1048713) | Cod sursa (job #2815144) | Cod sursa (job #3122946)
#include<bits/stdc++.h>
using namespace std;
ifstream in ("sdo.in");
ofstream out ("sdo.out");
int partition(vector<int>& arr, int left, int right, int pivotIndex) {
int pivotValue = arr[pivotIndex];
int storeIndex = left;
swap(arr[pivotIndex], arr[right]);
for (int i = left; i < right; i++) {
if (arr[i] < pivotValue) {
swap(arr[i], arr[storeIndex]);
storeIndex++;
}
}
swap(arr[storeIndex], arr[right]);
return storeIndex;
}
int quickSelect(vector<int>& arr, int left, int right, int k) {
if (left == right) {
return arr[left];
}
int pivotIndex = left + (right - left) / 2;
pivotIndex = partition(arr, left, right, pivotIndex);
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() {
int K, N;
vector<int> arr;
in >> N >> K;
for(int i = 0; i < N; i++){
int temp;
in >> temp;
arr.push_back(temp);
}
out << quickSelect(arr, 0, N-1, K-1);
return 0;
}