牛客网[最小的k个数]

题目描述

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

解法1；利用最大堆，O（nlogk）

vector<int> GetLeastNumbers_Solution(vector<int> input, int k) { if(input.empty()||k>input.size()||k<=0) return {}; vector<int>heap_input(input.begin(),input.begin()+k); make_heap(heap_input.begin(),heap_input.end()); for(int i=k;i<input.size();i++) { if(input[i]<*heap_input.begin()) { pop_heap(heap_input.begin(),heap_input.end()); heap_input.pop_back(); heap_input.push_back(input[i]); push_heap(heap_input.begin(),heap_input.end()); } } return heap_input; }

注意题目条件要考虑周期，对于k<=0这种情况开始没有考虑，导致在牛客网上编译通不过。

解法2：利用全排序  时间复杂度O（nlogn）

vector<int> GetLeastNumbers_Solution(vector<int> input, int k) { if(input.empty()||k>input.size()) return {}; vector<int>res; sort(input.begin(),input.end()); for(int i=0;i<k;i++) { res.push_back(input[i]); } return res; }