剑指offer 最小的k个数

一、相关定义

优先队列容器与队列一样，只能从队尾插入元素，从队首删除元素。但是它有一个特性，就是队列中最大的元素总是位于队首，所以出队时，并非按照先进先出的原则进行，而是将当前队列中最大的元素出队。这点类似于给队列里的元素进行了由大到小的顺序排序。元素的比较规则默认按元素值由大到小排序，可以重载“<”操作符来重新定义比较规则。

优先级队列可以用向量(vector)或双向队列(deque)来实现(注意list container不能用来实现queue，因为list的迭代器不是任意存取iterator，而pop中用到堆排序时是要求randomaccess iterator 的!)：
priority_queue<vector<int>, less<int> > pq1; 　　　 // 使用递增less<int>函数对象排序
priority_queue<deque<int>, greater<int> > pq2; 　　// 使用递减greater<int>函数对象排序
其成员函数有“判空(empty)” 、“尺寸(Size)” 、“栈顶元素(top)” 、“压栈(push)” 、“弹栈(pop)”等。

 

二、priority_queue

基本操作：

empty() 　　   如果队列为空，则返回真

pop()　　　　删除对顶元素，删除第一个元素

push() 　　     加入一个元素

size() 　　　  返回优先队列中拥有的元素个数

top() 　　　　返回优先队列对顶元素，返回优先队列中有最高优先级的元素

在默认的优先队列中，优先级高的先出队。在默认的int型中先出队的为较大的数。

头文件：

#include <queue>

声明方式：

1、普通方法：

priority_queue<int> q;   　　　　　　　　　　　　  //通过操作，按照元素从大到小的顺序出队

priority_queue<int,vector<int>, greater<int> > q;  　　//通过操作，按照元素从小到大的顺序出队

2、自定义优先级：

struct cmp {     

　　operator bool ()(int x, int y)     

　　{        

　　　　 return　x > y;　　 // x小的优先级高       //也可以写成其他方式，如： return p[x] > p[y];表示p[i]小的优先级高

　　}

};

priority_queue<int, vector<int>, cmp> q;    //定义方法

//其中，第二个参数为容器类型。第三个参数为比较函数。

 

此题直接sort完之后输出也可以，按照优先队列也可以。因为之前

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len=input.size();
        vector<int>ans;
        priority_queue <int,vector<int>,greater<int> > q;
        if(len<k)
            return ans;
        for(int i=0;i<len;i++)
            q.push(input[i]);
        for(int i=0;i<k;i++)
        {
            ans.push_back(q.top());
            q.pop();
        }
        return ans;   
            
    }
};

没做过优先队列相关的，所以在此记录。