ACdream 1099求第k大

瑶瑶的第K大

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

SubmitStatistic Next 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:

 1 /*************************************************************************
 2     > File Name: Kth.cpp
 3     > Author: Stomach_ache
 4     > Mail: sudaweitong@gmail.com
 5     > Created Time: 2014年08月02日 星期六 12时32分04秒
 6     > Propose: ACdream
 7  ************************************************************************/
 8 //输入挂+快速选择
 9 #include <cmath>
10 #include <string>
11 #include <cstdio>
12 #include <fstream>
13 #include <cstring>
14 #include <iostream>
15 #include <algorithm>
16 using namespace std;
17 
18 int n, k;
19 int a[5000002];
20 
21 int read() {
22       int x = 0;
23     char ch = ' ';
24     while (ch < '0' || ch > '9') ch = getchar();
25     while (ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar();
26     return x;
27 }
28 
29 int sort(int l, int r) {
30       if (l >= r) return a[l];
31       int pivot = a[(l+r)/2];
32     int i = l, j = r;
33     for ( ; ; ) {
34           while (i < j && a[i] <= pivot) i++; 
35         while (i < j && a[j] >= pivot) j--;
36         if (i < j) swap(a[i], a[j]);
37         else break;
38         i++; j--;
39     }
40     swap(a[i], a[(l+r)/2]);
41     if (i == k) return a[i];
42     if (i < k) return sort(i+1, r);
43     else return sort(l, i-1);
44 }
45 
46 int main(void) {
47       while (~scanf("%d %d", &n, &k)) {
48           for (int i = 1; i <= n; i++) 
49               a[i] = read();
50         k = n - k + 1;
51         int ans = sort(1, n);
52         printf("%d
", ans);
53     }
54     return 0;
55 }