ACdream OJ 1099 瑶瑶的第K大

瑶瑶的第K大

Time Limit: 10000/5000 MS (Java/Others)  Memory Limit: 512000/256000 KB (Java/Others)

Submit Statistic  Next Problem

Problem Description

一天，萌萌的妹子--瑶瑶(tsyao)很无聊，就来找你玩。可是你们都不知道玩什么。。。尴尬了一阵子，机智的瑶瑶就提议：“这样吧，你说N个整数xi，然后在随意说一个数字k，我能够快速地说出这些数字里面第k大的数字。”

 

Input

第1行 两个整数N，K以空格隔开；

第2行 有N个整数（可出现相同数字，均为随机生成），同样以空格隔开。

范围：n ≤ 5*10^6 , k ≤ n , 1 ≤ xi ≤ 10^8，单组输入，case数<=10。

Output

当有相同数字时，将数字按从大到小的顺序排好后，第k大的数字为有序数字的第k个。如2，2，1中三个数字中第一大数字为2，第二大数字也为2，第三大数字为1 。

Sample Input

5 2
5 4 1 3 1

Sample Output

4

直接sort一发，TLE，需要手写快排，快排每次需要找出一个枢纽，而这个枢纽位置的左边的数都不比枢纽大，右边的都不比枢纽小，所以这个我们可以在过程中判断答案是在枢纽左边、还是右边、还是就是枢纽。

#include <stdio.h>
#include <string.h>
#include <vector>
#include <math.h>
#include <stdlib.h>
#include <queue>
#include <algorithm>
using namespace std;
 
typedef long long ll;
#define rep(i,s,t) for(int i=s;i<t;i++)
#define red(i,s,t) for(int i=s-1;i>=t;i--)
#define ree(i,now) for(int i=head[now];i!=-1;i=edge[i].next)
#define clr(a,v) memset(a,v,sizeof a)
 
#define PI acos(-1.0)
#define eps 1e-6
 
inline int input(){
    int ret=0;char c=getchar();
    while(c<'0' || c>'9') c=getchar();
    while(c>='0' && c<='9'){
        ret=ret*10+c-'0';
        c=getchar();
    }
    return ret;
}
inline void output(int x){
    if(x<0){
        putchar('-');x=-x;
    }
    int len=0,data[10];
    while(x){
        data[len++]=x%10;x/=10;
    }
    if(!len)    data[len++]=0;
    while(len--)
        putchar(data[len]+48);
    putchar('\n');
}
 
const int N=5000005;
int n,k,a[N];
 
inline int part(int *a,int s,int t){
    int v=a[s];
    while(s<t){
        while(s<t && a[t]>=v) t--;
        if(s<t) a[s++]=a[t];
        while(s<t && a[s]<=v) s++;
        if(s<t) a[t--]=a[s];
    }
    a[s]=v;
    return s;
}
 
inline int Sort(int *a,int s,int t,int k){
    if(s<t){
        int p=part(a,s,t);
        if(p==k) return a[p];
        else if(p>k) return Sort(a,s,p-1,k);
        else return Sort(a,p+1,t,k);
    }
    else return a[s];
}
 
int main(){
    n=input(),k=input();
    rep(i,0,n) a[i]=input();
    output(Sort(a,0,n-1,n-k));
    return 0;
}