题目描述
求第k大的数
给定一个长度为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
这道题用的快速排序,代码有参考别人https://blog.csdn.net/weixin_44774255/article/details/107887995
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <ctime>
using namespace std;
const int maxn = 1000010;
int a[maxn];
int randpartition(int a[], int left, int right) {
int temp = a[left];
while (left < right) {
if (left<right && a[right]>right) right--;
a[left] = a[right];
if (left<right && a[left]>right) left++;
a[right] = a[left];
}
a[left] = temp;
return left;
}
int randselect(int a[], int left, int right, int k) {
if (left == right) return a[left];
int p = randpartition(a, left, right);
//划分后主元的位置为p
int m = p - left + 1;
if (k == m) return a[p];
if (k < m)
return randselect(a, left, p - 1, k);
else {
return randselect(a, p + 1, right, k - m);
}
}
int main() {
int k, n;
while (scanf("%d%d",&n,&k) != EOF) {
for (int i = 0;i < n;i++) {
scanf("%d", &a[i]);
}
int x = randselect(a, 0, n - 1, n - k + 1);
printf("%d\n", x);
}
return 0;
}