题目链接:http://codeup.cn/problem.php?cid=100000587&pid=0
题目描述
给定一个长度为n(1≤n≤1,000,000)的无序正整数序列,以及另一个数k(1≤k≤1,000,000)(关于第k大的数:例如序列{1,2,3,4,5,6}中第3大的数是4。)
输入
第一行两个正整数m,n。
第二行为n个正整数。
输出
第k大的数。
样例输入
6 3
1 2 3 4 5 6
样例输出
4
代码
#include<stdio.h>
const int maxn = 1000010;
int a[maxn], n, k;
int Partition(int a[], int left, int right) {
int temp = a[left];
while(left < right) {
while(left < right && a[right] > temp)
right --;
a[left]=a[right];
while(left < right && a[left] <= temp)
left ++;
a[right]=a[left];
}
a[left] = temp;
return left;
}
int Slect(int a[], int left, int right, int k) {
if(left == right)
return a[left];
int p = Partition(a, left, right);
int m = p - left +1;
if(k == m)
return a[p];
if(k < m)
return Slect(a, left, p - 1, k);
else
return Slect(a, p + 1, right, k - m);
}
int main () {
while(scanf("%d%d", &n, &k) != EOF){
for(int i = 0; i < n; i++)
scanf("%d", &a[i]);
printf("%d\n", Slect(a, 0, n - 1, n - k +1)); //第k大的数可看成是第 n-k+1 小
}
return 0;
}