瑶瑶的第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 1Sample Output
4Hint
如2,2,1中三个数字中第一大数字为2,第二大数字也为2,第三大数字为1 。
由于n过大,需要使用输入挂,然后O(n)的快速选择即可。1A
Accepted Code:
1 /************************************************************************* 2 > File Name: Kth.cpp 3 > Author: Stomach_ache 4 > Mail: [email protected] 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\n", ans); 53 } 54 return 0; 55 }
ACdream 1099求第k大
时间: 2024-11-08 18:29:41