Pagini recente » Cod sursa (job #20283) | Cod sursa (job #1634758) | Cod sursa (job #433404) | Cod sursa (job #1445023) | Cod sursa (job #2622690)
#include <iostream>
#include <vector>
#include <fstream>
using namespace std;
ifstream fin("sdo.in");
ofstream fout("sdo.out");
int n, k;
vector<int> v;
int partition(int left, int right)
{
int i = left;
int pivot = rand() % (right - left + 1) + left;
swap(v[right], v[pivot]);
while(left < right)
{
if(v[left] <= v[right])
swap(v[i++], v[left]);
left++;
}
swap(v[i], v[right]);
return i;
}
// partition
int kthSmallest(int left, int right, int k)
{
if(k > 0 && k <= right - left + 1)
{
int index = partition(left, right);
if(index - left == k - 1)
return v[index];
if(index - left > k - 1)
return kthSmallest(left, index - 1, k);
return kthSmallest(index + 1, right, k - index + left - 1);
}
return 0;
}
int main()
{
fin >> n >> k;
for(int i = 0; i < n; ++i)
{
int x;
fin >> x;
v.push_back(x);
}
fout << kthSmallest(0, n - 1, k);
v.clear();
return 0;
}