C++面试——最小的K个数（LeetCode 面试题40）

最小的K个数

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

示例

**输入：**arr = [0,1,2,1], k = 1
输出：[1,2] 或者 [2,1]
快速排序

class Solution {
public:
    int Paration(vector<int>&arr,int low,int high){
        int pivot = arr[low];
        while(low<high){
            if(low<high&&arr[high]>=pivot){
                --high;
            }
            arr[low]=arr[high];
            if(low<high&&arr[low]<=pivot){
                ++low;
            }
            arr[high]=arr[low];
        }
        arr[low]=pivot;
        return low;
    } 
    void QuickSort(vector<int>&arr,int low ,int high){
        if(low<high){
            int pivotpos = Paration(arr,low,high);
            QuickSort(arr,low,pivotpos-1);
            QuickSort(arr,pivotpos+1,high);
        }
    }

    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> ans;
        QuickSort(arr,0,arr.size()-1);
        for(int i=0;i<k;i++){
            ans.push_back(arr[i]);
        }
        return ans;
    }
};

优先级队列

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        priority_queue<int,vector<int>, greater<int>> q;
		for (auto a : arr) {
			q.push(a);
		}
		vector<int> ans;
		for (int i = 0; i < k; i++) {
			ans.push_back(q.top());
			q.pop();
		}
		return ans;
    }
};