《算法笔记》Codeup练习 4.7小节 其他高技巧与算法

问题A 找第K大数

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

注意
这题可以选择把数字排序之后，输出第k个数字（这样也是可以通过的）
不过有更有的解法——随机选择。有个快排的思想在里面。

#include<stdio.h>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<math.h>
#include<algorithm>
#include<stdlib.h>
#include<ctime>
#include<time.h>
#include<cstdlib>
#define MAX 1000001
using namespace std;

int num[MAX];
/*
bool cmp(int a,int b){
    return a>b;
}
*/
int randPar(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){
        return;
    }
    int p=randPar(A,left,right);
    int M=p-left+1;
    if(K==M)
        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,k;
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++)
        scanf("%d",&num[i]);
    //sort(num,num+n,cmp);
    randSelect(num,0,n-1,k);
    printf("%d\n",num[n-k]);
    return 0;
}