问题 A: 求第k大数

题目链接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;
}
发布了148 篇原创文章 · 获赞 7 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Rhao999/article/details/104155924
今日推荐