Q40数组中最小的k个数

数组中最小的k个数

题目

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

思路

思路1：

利用 快排的 partition 函数对 数组进行初步排序，使得位于 第 k-1前后的数组是排序的。

输出前k个数字。

思路2：

处理海量数据时，可以使用一个k个元素的 multiset，顶上放最大的数字，不断判断是否更新set里面的数字。

实现

class Solution {
public:
    int partition(vector<int>& numbers, int left, int right)
    {
        if(numbers.empty() || left<0 || right>=numbers.size() || left>right)
            return -1;
        int pivot = numbers[left];
        while(left<right)
        {
            while(left<right && numbers[right]>=pivot)
                --right;
            numbers[left] = numbers[right];
            while(left<right && numbers[left]<=pivot)
                ++left;
            numbers[right] = numbers[left];
        }
        numbers[left] = pivot;
        return left;
    }
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> result;
        if(input.empty() || k<=0 || input.size()<k)
            return result;
        int left = 0;
        int right = input.size()-1;  //记得是减1
        int pindex = partition(input, left, right);
        int mark = k-1;
        while(pindex!=mark)
        {
            if(pindex<mark)
            {
                left = pindex+1;
                pindex = partition(input, left, right);
            }
            else
            {
                right = pindex-1;
                pindex = partition(input, left, right);
            }
        }
        for(int i=0; i<k; ++i)
            result.push_back(input[i]);
        return result;
    }
};