【经典算法】第k小的数

快排应用的经典题型，看该题之前请先看快速排序，附带链接：快速排序

题目简述

给出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);
}