快排应用的经典题型,看该题之前请先看快速排序,附带链接:快速排序
题目简述
给出n个数,找出n个数排序后第k个数,即找出第k小的数。
输入格式
第一行包含两个整数 n 和 k。
第二行包含 n 个整数(所有整数均在1~1e9范围内),表示整数数列。
输出格式
输出一个整数,表示数列的第k小数。
输入样例:
5 3
2 4 1 5 3
输出样例:
3
利用快速排序的性质
该题是利用快速排序中的一个性质,即将主元放在正确的位置,使得左边的数都比主元小,右边的数比主元大,这样会产生一个性质:该数后面是不会移动的,即该数落在排序后的正确位置,也就是序列中第k小的数。
利用上面的性质,我们在排序的过程中,判断我们需要找的第k个数在主元的左边还是右边(二分的思想),如果第k个数在主元的左边,那么只需要递归左边即可。
时间复杂度O(N)
AC代码
int findKth(int a[], int l ,int r,int k){
if (l == r) return a[l];
int x = a[l + r >> 1] , i = l - 1 , j = r + 1;
while (i < j){
while (a[++i] < x);
while (a[--j] > x);
if (i < j) swap(a[i] , a[j]);
}
//计算主元左边数字个数
int lcnt = j - l + 1;
//判断第k个数在左边还是右边
if (k <= lcnt) return findKth(a, l, j, k);
else return findKth(a, j + 1, r, k - lcnt);
}