问题 A: 求第k大数

题目描述
求第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;
}