40_最小的k个数

最小的k个数

输入整数数组 arr ，找出其中最小的 k 个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。

示例 1：

输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]
示例 2：

输入：arr = [0,1,2,1], k = 1
输出：[0]

限制：

• 0 <= k <= arr.length <= 10000
• 0 <= arr[i] <= 10000

分情况讨论： 1-》 当数组的大小较小，并行可以进行改变是，可以使用快速排序的方法，找对第k大的数，这个时候前面的K个数就是较小的k个数。
2-》 当原数组不可改变，且数组非常大的时候，方法一不在适用，可以使用一个堆来保存前k个大小的数，实现动态比较保存。

解法一(改变原数组顺序)：

class Solution {
public:
vector<int> getLeastNumbers(vector<int>& arr, int k) {
        if(arr.empty() || (arr.size() < k || k <= 0))
            return vector<int>(0,0);
        int low = 0;
        int high = arr.size() - 1;
        int index = partition(arr,low,high);
        k = k -1;
        while(index != k){
            if(index < k){
                low = index + 1;
                index = partition(arr,low,high);
            } else{
                high = index - 1;
                index = partition(arr,low,high);
            }
        }
        return vector<int>(arr.begin(),arr.begin() + index+1);
    }
    int partition(vector<int>& nums,int low,int high){
        int pivot = nums[low];
        int i = low;
        int j = high;
        while(i != j){
            while(i < j & nums[j]>pivot)
                j--;
            if(i < j)
                nums[i++] = nums[j];
            while(i< j & nums[i] < pivot)
                i++;
            if(i < j)
                nums[j--] = nums[i];
        }
        nums[i] = pivot;
        return i;
    }
};

可以使用一个大小为k的大顶堆来存放K个数，当堆不满的时候，直接放入；当堆满时，将数与堆顶元素进行比较，如果来的数字比堆顶元素要小，删除堆顶元素，插入新数，否则抛弃。
知识点： 利用multiset实现堆排序，multiset erase函数，insert函数的使用

解法二(利用STL中multiset使用红黑树实现的性质 实现大顶堆)

class Solution {
public:
    typedef multiset<int,greater<int>> intSet;
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        if(arr.empty() || (arr.size() < k || k <= 0))
            return vector<int>(0,0);
        intSet result;
        for(auto i:arr){
            if(result.size() < k)
                result.insert(i);
            else if(*result.begin() > i){
                result.erase(result.begin());
                result.insert(i);
            }
        }
        return vector<int>(result.begin(),result.end());
    }
};