题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路1:冒泡排序O(n*k)
当然,如果是海量数据,那么此时效率太低
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> res;
if(k > input.size() || k <= 0)
return res;
for(int i = input.size();i > input.size() - k;i--){
int min = 0;
for(int j = 0;j < i;j++){
if(input[min] < input[j]){
swap(input[min],input[j]);
}
min = j;
}
res.push_back(input[min]);
}
return res;
}
};
思路2:堆排序O(nlogk)
先把这n个数中的前k个输入进大顶堆
然后遍历后面的数,
如果比堆顶小,就删除堆顶,并把这个数入堆
否则继续
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> res;
if(k > input.size() || k <= 0)
return res;
priority_queue<int> q;
for(int i = 0;i < input.size();i++){
if(i < k){
q.push(input[i]);
}else{
if(input[i] < q.top()){
q.pop();
q.push(input[i]);
}
}
}
for(int i = 0;i < k;i++){
res.push_back(q.top());
q.pop();
}
return res;
}
};