POJ 4087 数据筛选(快排)(第k小的数)

编程题＃2. 数据筛选
注意： 总时间限制: 10000ms 内存限制: 3000kB

AC

描述
小张需要从一批数量庞大的正整数中挑选出第k小的数，因为数据量太庞大，挑选起来很费劲，希望你能编程帮他进行挑选。

输入
第一行第一个是数据的个数n(10<=n<=106)，第二个是需要挑选出的数据的序号k(1<=k<=105)，n和k以空格分隔；

第二行以后是n个数据T(1<=T<=109)，数据之间以空格或者换行符分隔。

输出
第k小数（如果有相同大小的也进行排序，例如对1,2,3,8,8，第4小的为8，第5小的也为8）。

样例输入

10 5
1 3 8 20 2
9 10 12 8 9

样例输出

8

代码

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

vector<int> vec;

int Partition(int l, int r) {
	int temp = vec[r];
	while (l != r) {
		while (vec[l] <= temp && r > l)
			++l;
		if (l < r) { vec[r] = vec[l]; --r; }

		while (vec[r] >= temp && r > l)
			--r;
		if (l < r) { vec[l] = vec[r]; ++l; }
	}
	vec[l] = temp;
	return l;
}

void QuickSort(int l, int r) {
	if (r <= l) { return; }
	int pivot = (l + r) / 2;
	swap(vec[pivot], vec[r]);
	pivot = Partition(l, r);
	QuickSort(l, pivot - 1);
	QuickSort(pivot + 1, r);
}

int main() {
	int n, k, num, n1, n2;
	cin >> n >> k;

	n1 = n;
	for (int i = 0; i <= n / 100000; ++i) {
		n2 = n1 > 100000 ? 100000 : n1;
		for (int j = 0; j < n2; ++j) {
			cin >> num;
			vec.push_back(num);
		}
		QuickSort(0, (int)vec.size() - 1);
		vec.erase(vec.begin() + k, vec.end());
		n1 -= n2;

		if (n1 <= 0) {
			cout << vec[k - 1];
			return 0;
		}
	}
}

Github: Westwood2723
Weibo: Cecilia_西木