剑指offer 最小的k个数

1.题目描述

输入n个数字，找出其中最小的k个数。

注：有的还会要求保持原有的顺序输出

2.分析

两种思路：
1）利用快排的思想，找到第k大的分界点。左边的元素都是比它小的
2）维护一个长度为k的容器，里面存放当前最小的k个数，遍历一个新的元素判断是否将其插入该容器。

3.代码

快排的思想，复杂度O(n)

void LeastK(int* array, int n,int k) {    if (array == NULL || k <= 0 || k >= n)        return;    int left = 0;    int right = n - 1;    int pivot = mypartition(array, 0, n - 1);    while (pivot != k - 1) {        if (pivot > k - 1) {            right = pivot - 1;            pivot = mypartition(array, left, right);        }        else {            left = pivot + 1;            pivot = mypartition(array, left, right);        }           }    for (int i = 0; i < k; i++)        cout << array[i] << " ";    cout << endl;}

使用最大堆,复杂度nlog(k)

void LeastK(int* array, int n, int k) {    if (array == NULL || k <= 0 || k >= n)        return;    priority_queue<int, vector<int>> maxHeap;//默认就是最大堆    for (int i = 0; i < n; i++) {        if (maxHeap.size() < k)            maxHeap.push(array[i]);        else if (array[i] < maxHeap.top()) {            maxHeap.pop();            maxHeap.push(array[i]);        }    }    while (!maxHeap.empty())    {        cout << maxHeap.top() << " ";        maxHeap.pop();    }    cout << endl;}

或者可以用multiset，降序排列，取set第一个元素是最大的。

#include<functional>multiset<int, greater<int>> myset;