编程题#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_西木