Leetcode之 最小的k个数

题目描述

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

思路

1. 哈希表，题目中给定arr.length<10000,arr[i]<10000，则建立一个Hash数组记录每个数字出现的次数，之后从Hash[0]开始输出，若Hash[i]>0，则循环输出直至Hash[i]=0 或 k=0
2. 堆的思想

代码

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;
    }
};