算法笔记4.7--求第K大的数

给定一个长度为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>
#include <cstdlib>
#include <ctime>
#include <algorithm>
using namespace std;

const int maxn = 1000010;
int A[maxn];

int randPartition(int A[],int left,int right){
    int p = round(1.0*rand() / RAND_MAX*(right - left) + left);
    swap(A[p],A[left]);
    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 randSelect(int A[],int left,int right,int K){
    if(left == right){
        printf("%d\n",A[left]);
        return;
    }
    int p = randPartition(A,left,right);
    int M = p - left + 1;
    if(K == M){
        printf("%d\n",A[p]);
        return;
    }
    if(K < M){
        randSelect(A,left,p-1,K);
    }
    else
    {
        randSelect(A,p+1,right,K-M);
    }
}

int main(){
    srand((unsigned)time(NULL));
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++){
        scanf("%d",A+i);
    }
    randSelect(A,0,n-1,n-m+1);
    return 0;
}

AC