ACdream 1099求第k大

题目链接

瑶瑶的第K大

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

SubmitStatisticNext Problem

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 。

由于n过大，需要使用输入挂，然后O(n)的快速选择即可。1A

Accepted Code:

 /*************************************************************************
     > File Name: Kth.cpp
     > Author: Stomach_ache
     > Mail: sudaweitong@gmail.com
     > Created Time: 2014年08月02日 星期六 12时32分04秒
     > Propose: ACdream
  ************************************************************************/
 //输入挂+快速选择
 #include <cmath>
 #include <string>
 #include <cstdio>
 #include <fstream>
 #include <cstring>
 #include <iostream>
 #include <algorithm>
 using namespace std;
 
 int n, k;
 int a[];
 
 int read() {
       int x = ;
     char ch = ' ';
     while (ch < '' || ch > '') ch = getchar();
     while (ch >= '' && ch <= '') x = x *  + ch - '', ch = getchar();
     return x;
 }
 
 int sort(int l, int r) {
       if (l >= r) return a[l];
       int pivot = a[(l+r)/];
     int i = l, j = r;
     for ( ; ; ) {
           while (i < j && a[i] <= pivot) i++;
         while (i < j && a[j] >= pivot) j--;
         if (i < j) swap(a[i], a[j]);
         else break;
         i++; j--;
     }
     swap(a[i], a[(l+r)/]);
     if (i == k) return a[i];
     if (i < k) return sort(i+, r);
     else return sort(l, i-);
 }
 
 int main(void) {
       while (~scanf("%d %d", &n, &k)) {
           for (int i = ; i <= n; i++)
               a[i] = read();
         k = n - k + ;
         int ans = sort(, n);
         printf("%d\n", ans);
     }
     return ;
 }