找第k小的数

设计一个平均时间为O(n)的算法，在n(1<=n<=1000)个无序的整数中找出第k小的数。

提示：函数int partition(int a[],int left,int right)的功能是根据a[left]~a[right]中的某个元素x（如a[left])对a[left]~a[right]进行划分，划分后的x所在位置的左段全小于等于x,右段全大于等于x,同时利用x所在的位置还可以计算出x是这批数据按升非降序排列的第几个数。因此可以编制int find(int a[],int left,int right,int k)函数，通过调用partition函数获得划分点，判断划分点是否第k小，若不是，递归调用find函数继续在左段或右段查找。

输入格式:

输入有两行：

第一行是n和k，0<k<=n<=10000

第二行是n个整数

输出格式:

输出第k小的数

输入样例:

在这里给出一组输入。例如：

10 4
2 8 9 0 1 3 6 7 8 2

输出样例:

在这里给出相应的输出。例如：

2

代码：

#include<iostream>
using namespace std;

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);
}
int main(){
    int n, k;
    cin >> n >> k;
    int a[n];
    for(int i=0; i<n; i++)
        cin >> a[i];
    cout << findKth(a, 0, n, k);
    return 0;
}


// https://blog.csdn.net/Miiiiiiiiiii/article/details/110006588?spm=1001.2101.3001.6661.1&utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-1-110006588-blog-127938630.pc_relevant_aa&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-1-110006588-blog-127938630.pc_relevant_aa&utm_relevant_index=1