[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;

}