瑶瑶的第K大
Time Limit: 10000/5000 MS (Java/Others)
Memory Limit: 512000/256000 KB (Java/Others)
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;
}