[ACdream 1099] 瑶瑶的第K大

瑶瑶的第K大

Time Limit: 4000/2000MS (Java/Others) Memory Limit: 256000/128000KB (Java/Others)

Problem Description

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

Input

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

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

0 < n ≤ 5*10^6 , 0 < k ≤ n

1 ≤ xi ≤ 10^8

Output

输出第 k 大的数字。

Sample Input

5 2
5 4 1 3 1

Sample Output

4

Hint

如2，2，1中三个数字中第一大数字为2，第二大数字也为2，第三大数字为1 。

 

这题卡时间卡得有点紧、利用快排的思想处理、加输入外挂3964ms飘过= =

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 6000010

int n,k;
int a[N];

int input()
{
    int ans=0;
    char a;
    while((a=getchar())<'0'||a>'9');
    ans=a-'0';
    while((a=getchar())>='0'&&a<='9')
    {
        ans=ans*10+a-'0';
    }
    return ans;
}
int qsort(int s,int t)
{
    int i=s,j=t,tmp;
    if(s<t)
    {
        tmp=a[s];
        while(i!=j)
        {
            while(j>i && a[j]<=tmp) j--;
            a[i]=a[j];
            while(i<j && a[i]>=tmp) i++;
            a[j]=a[i];
        }
        a[i]=tmp;
        if(k==i) return a[i];
        else if(k<i) return qsort(s,i-1);
        else return qsort(i+1,t);
    }
}
int main()
{
    n=input();
    k=input();
    for(int i=1;i<=n;i++)
        a[i]=input();
    printf("%d\n",qsort(1,n));
    return 0;
}