作业训练二编程题11. 最小的K个数

【问题描述】

输入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;
}