【问题描述】
输入n个整数,找出其中最小的k(k<=n)个不同数。例如输入4,5,1,6,1,7,3,8这8个数字,则最小的4个数字是1,3,4,5。
【输入形式】
每个测试案例包括2行:
第一行为2个整数n,k(1<=n,k<=200000),表示数组的长度。
第二行包含n个整数,表示这n个数,数组中的数的范围是[0,1000 000 000]。
【输出形式】
对应每个测试案例,输出最小的k个数,并按从小到大顺序打印(如果不存在k个不同的数,则按照实际数量进行输出)。
【样例输入】
8 4 4 5 1 6 2 7 3 8
【样例输出】
1 2 3 4
【训练提示】
1、数的范围从0到1000000000,使用数组记录那些数出现过就不是太合适
2、需要去除重复的数,需要从小到大排序----set就是一个不错的选择
【思路分析】
本题为LeetCode 面试题 17.14. 最小K个数,做法较多,可以直接排序(时间复杂度为)也可以用堆排(时间复杂度为
)。接下来介绍数组划分方法。
其思想类似于快排,快排的思想是通过基准值pivot调整数组,pivot前的所有元素严格小于pivot,pivot后的所有元素严格大于pivot。此处我们加以限制pivot的下标,使pivot为数组的第k个元素。那么包含pivot的前k个元素即为所求答案。
要想找到pivot,即为查找第k小的元素。对于每个划分区间,将该区间的前k个元素置于左端。
提供的代码的时间复杂度:期望为,最坏情况下为
。证明过程比较复杂,详情请看《算法导论》第 9 章第 2 小节。
#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
int partition(vector<int> &nums, int l, int r) {
int pivot = nums[r];
int i = l - 1;
for (int j = l; j <= r - 1; ++j) {
if (nums[j] <= pivot) {
i = i + 1;
swap(nums[i], nums[j]);
}
}
swap(nums[i + 1], nums[r]);
return i + 1;
}
// 基于随机的划分
int randomized_partition(vector<int> &nums, int l, int r) {
int i = rand() % (r - l + 1) + l;
swap(nums[r], nums[i]);
return partition(nums, l, r);
}
void randomized_selected(vector<int> &arr, int l, int r, int k) {
if (l >= r) return;
int pos = randomized_partition(arr, l, r);
int num = pos - l + 1;
if (k == num) return;
else if (k < num)randomized_selected(arr, l, pos - 1, k);
else randomized_selected(arr, pos + 1, r, k - num);
}
int main() {
int n, k, tmp;
cin >> n >> k;
set<int> initialData;
vector<int> arr;
for (int i = 0; i < n; ++i) {
cin >> tmp;
initialData.emplace(tmp);
}
for (const auto &item: initialData) arr.emplace_back(item);
if (k > arr.size()) k = arr.size();
srand((unsigned) time(nullptr));
randomized_selected(arr, 0, arr.size() - 1, k);
sort(arr.begin(), arr.begin() + k);
for (int i = 0; i < k; ++i) cout << arr[i] << " ";
return 0;
}