1. 程式人生 > >【牛客】最小的k個數

【牛客】最小的k個數

使用topK的方法,如果不想寫堆的話,瞭解幾個庫裡的堆的函式

std::make_heap將[start, end)範圍進行堆排序,預設使用less, 即最大元素放在第一個。
std::pop_heap將front(即第一個最大元素)移動到end的前部,同時將剩下的元素重新構造成(堆排序)一個新的heap。
std::push_heap對剛插入的(尾部)元素做堆排序。
std::sort_heap將一個堆做排序,最終成為一個有序的系列,可以看到sort_heap時,必須先是一個堆(兩個特性:1、最大元素在第一個 2、新增或者刪除元素以對數時間),因此必須先做一次make_heap.

思路:
1、引數檢驗:len<=0、k>len、k<=0 都是不對的
2、建堆:使用make_heap建立k個節點的最大堆
3、使用剩餘元素和堆頂元素比較,
4、如果小於堆頂元素,那麼刪除堆頂元素,將此元素插入
5、最後使用sort_heap排序,輸出

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        //topK
        int len=input.size();
        if(len<=0||k>len||k<=0)
            return vector<int>();
        //建立k個元素的堆(大堆)
        vector<int> res;
        for(int i=0
;i<k;i++){ res.push_back(input[i]); } make_heap(res.begin(),res.end()); for(int i=k;i<len;i++){ if(res[0]>input[i]){ pop_heap(res.begin(),res.end());//將待刪除元素放到最後 res.pop_back(); res.push_back(input[i]); push_heap(res.begin(),res.end()); } } //最後,再從小到大輸出
sort_heap(res.begin(),res.end()); return res; } };