算法1
时间复杂度 $O(nlgn)$
C++ 代码
#include <iostream>
using namespace std;
const int N = 1e6;
int n, k, q[N];
void quick_sort(int q[], int l, int r)
{
if(l >= r) return;
int i = l - 1, j = r + 1, x = q[l + r >> 1];
while(i < j)
{
do i ++; while(q[i] < x);
do j --; while(q[j] > x);
if(i < j) swap(q[i], q[j]);
}
quick_sort(q, l, j);
quick_sort(q, j + 1, r);
}
int main()
{
cin >> n >> k;
for(int i = 0; i < n; i ++) cin >> q[i];
quick_sort(q, 0, n - 1);
cout << q[k - 1] << endl;
return 0;
}
算法2
时间复杂度 $O(n)$
C++ 代码
#include <iostream>
using namespace std;
const int N = 1e6;
int n, q[N];
int quick(int l, int r, int k)
{
if(l == r) return q[l];
int i = l - 1, j = r + 1, x = q[l + r >> 1];
while(i < j)
{
do i ++; while(q[i] < x);
do j --; while(q[j] > x);
if(i < j) swap(q[i], q[j]);
}
// l......j, j+1......r
int lcnt = j - l + 1;
if(lcnt >= k) return quick(l, j, k);
else return quick(j + 1, r, k - lcnt);
}
int main()
{
int k;
cin >> n >> k;
for(int i = 0; i < n; i ++) cin >> q[i];
cout << quick(0, n - 1, k);
return 0;
}