原题链接 第k个数
解题思路:
利用分治和归并的思想,先利用类似快速排序法的思路,选择一个标兵,然后将数组中大于标兵的数全部放在右边,小于标兵的数目全部放在左边,然后计算两边各自的长度,如果左边的长度大于k,那么第k个数一定在左边,对左边的数组进行相同的操作,也就是递归,如果左边的长度小于k,那么就递归右边。
#include<iostream>
using namespace std;
const int maxn = 100000 + 5;
int num[maxn];
int quick_sort(int l, int r, int k)
{
if (l == r)
{
return num[l];
}
else
{
int i = l - 1, j = r + 1, x = num[l];
while (i < j)
{
do ++i; while (num[i] < x);
do --j; while (num[i] > x);
if (i < j)
{
swap(num[i], num[j]);
}
}
int s1 = j - l + 1;
if (k<=s1)
{
return quick_sort(l, j, k);
}
return quick_sort(j+1, r, k - s1);
}
}
int main(void)
{
int n, k;
scanf("%d %d", &n, &k);
for (int i = 0; i < n; i++)
{
scanf("%d", &num[i]);
}
printf("%d", quick_sort(0, n - 1, k));
return 0;
}