求第k大数

题目描述

求第k大的数

(Kth.pas/c/cpp)

给定一个长度为n（1≤n≤1,000,000）的无序正整数序列，以及另一个数k（1≤k≤1,000,000）（关于第k大的数：例如序列{1,2,3,4,5,6}中第3大的数是4。）

输入

第一行两个正整数m,n。

第二行为n个正整数。

输出

第k大的数。

样例输入

6 3
1 2 3 4 5 6

样例输出

4

提示

#include <cstdio>
const int maxn=1e6+10;
int a[maxn];

int Partition(int A[],int left,int right){
    int temp=A[left];
    while(left<right){
        while(left<right&&A[right]>temp) right--;
        A[left]=A[right];
        while(left<right&&A[left]<=temp) left++;
        A[right]=A[left];
    }
    A[left]=temp;
    return left;
}

void quick_sort(int A[],int left,int right){
    if(left<right){
        int pos=Partition(A,left,right);
        quick_sort(A,left,pos-1);
        quick_sort(A,pos+1,right);
    }
}

int main(){
    int n,k;
    while(scanf("%d%d",&n,&k)==2){
        for(int i=0;i<n;i++) scanf("%d",&a[i]);
        int q=Partition(a,0,n-1);
        if(n-1-q>=k) quick_sort(a,q+1,n-1);
        else quick_sort(a,0,q);
        printf("%d\n",a[n-k]);
    }
    return 0;
}