找第k小的数(快速排序法)

7-2 找第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<bits/stdc++.h>
using namespace std;

int jizhun(int *a,int left,int right){
    int i=left;  //基准
    int j=right+1;
    while(i<j){
        while(i<=right && a[++i]<a[left]);
        while(a[--j]>a[left]);
        if(i>=j) break;
        swap(a[i],a[j]);
    }
    swap(a[left],a[j]);
    return j;
}

void quicksort(int *a,int left,int right){
    if(left>=right) return;
    int p=jizhun(a,left,right);
    quicksort(a,left,p-1);
    quicksort(a,p+1,right);
}

int main(){
    int a[10005];
    int n,k;
    cin>>n>>k;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    quicksort(a,0,n-1);
    cout<<a[k-1];
    return 0;
}