题目描述
输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
思路
- 哈希表,题目中给定arr.length<10000,arr[i]<10000,则建立一个Hash数组记录每个数字出现的次数,之后从Hash[0]开始输出,若Hash[i]>0,则循环输出直至Hash[i]=0 或 k=0
- 堆的思想
代码
class Solution {
public:
vector<int> getLeastNumbers(vector<int>& arr, int k) {
vector<int> res;
if(arr.size()==0 || k==0)
return res;
vector<int> Hash(10000);
for(int i = 0;i<arr.size();i++)
{
Hash[arr[i]]++;
}
int count = 0;
while(k>0)
{
while(Hash[count]>0 && k > 0)
{
res.push_back(count);
Hash[count]--;
k--;
}
count++;
}
return res;
}
};
方法二:
class Solution {
public:
vector<int> getLeastNumbers(vector<int>& arr, int k) {
vector<int> res;
if(k == 0){
return res;
}
priority_queue<int> h;
for(int num : arr){
if(h.size() < k){
h.push(num);
}else{
if(h.top() <= num){
continue;
}else{
h.pop();
h.push(num);
}
}
}
while(!h.empty()){
res.push_back(h.top());
h.pop();
}
return res;
}
};