求第k大的数

问题 A: 求第k大数 - Codeup新家 (hustoj.com)http://codeup.hustoj.com/problem.php?cid=100000587&pid=0#

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<ctime>
using namespace std;
const int maxn=1000010;
int nums[maxn]={0};
int partition(int i, int j){
    int pos = (int)(((double)rand()/RAND_MAX * (j-i))+i);
    swap(nums[i], nums[pos]);
    int target = nums[i];
    int l=i,r=j;
    while(l<r){
        while(l<r&&nums[r]>target)
            r--;
        if(l<r){
            nums[l]=nums[r];
            l++;
        }
        while(l<r&&nums[l]<=target) //为什么不能为等号
            l++;
        if(l<r){
            nums[r]=nums[l];
            r--;
        }
    }
    nums[l]=target;
    return l;
}
//在nums[i]~nums[j]中，找到第k小的值
int selectKMin(int i, int j, int k){
    if(i==j) return nums[i];
    int pos = partition(i, j);
    //printf("%d %d %d %d:",i,j,pos,k);// 4 3 4 1死循环
    if(pos==k) return nums[k];
    if(pos<k) return selectKMin(pos+1,j,k-pos);
    return selectKMin(i, pos-1, k);
}
int main(){
    srand((unsigned)time(NULL)); // 随机数
    int n, m;
    while(scanf("%d %d", &n, &m)!=EOF){
        for(int i=0;i<n;i++)
            scanf("%d" ,&nums[i]);
        int k = n-m;
        int ans = selectKMin(0,n-1,k);
        printf("%d\n", ans);
    }
    
}