[算法系列之四]优先级队列

简介:

【概念】

优先级队列,顾名思义,就是一种根据一定优先级存储和取出数据的队列。它可以说是队列和排序的完美结合体,不仅可以存储数据,还可以将这些数据按照我们设定的规则进行排序。优先级队列是堆的一种常见应用。有最大优先级队列(最大堆)和最小优先级队列(最小堆)。优先级队列是一种维护有一组元素构成的集合S的数据结构。

【优先队列支持的基本运算】

[cpp]  view plain copy
  1. //建立一个保存元素为int的优先级队列,其实是建了一个小顶堆  
  2. //但是请特别注意这样的建的堆默认是大顶堆,即我们从堆顶去的元素是整个堆中元素最大的。  
  3.   
  4. priority_queue<int> Heap;  
  5.   
  6. //可以这样建一个表示小顶堆的优先级队列  
  7.   
  8. priority_queue<int , vector<int>, greater<int> > Heap;  
  9.   
  10. //将元素x放入优先级队列中  
  11.   
  12. Heap.push(x);  
  13.   
  14. //取出优先级队列第一个元素(堆顶元素),保存在x中  
  15.   
  16. int x = Heap.top();  
  17.   
  18. //弹出堆顶元素,取出后堆会自动调整为一个最小堆(最大堆)  
  19.   
  20. Heap.pop();  
  21.   
  22. //判断是否为空  
  23.   
  24. Heap.empty();  
  25.   
  26. //头文件  
  27.   
  28. #include<queue>  

【原理】

priority_queue调用 STL里面的 make_heap(), pop_heap(), push_heap() 算法实现,也算是堆的另外一种形式。

用make_heap(), pop_heap(), push_heap() 简单实现一个最大优先级队列

/*********************************
*   日期:2015-01-06
*   作者:SJF0115
*   题目: 简单实现最大优先级队列
*   博客:
**********************************/
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

//简单实现最大优先级队列
template<typename T>
class priority_queue{
private:
    // 数据
    vector<T> data;
public:
    // 进队列
    void push(T val){
        data.push_back(val);
        push_heap(data.begin(),data.end());
    }
    // 出队列
    void pop(){
        pop_heap(data.begin(),data.end());
        data.pop_back();
    }
    // 头元素
    T top(){
        return data.front();
    }
    // 大小
    int size(){
        return data.size();
    }
    // 是否为空
    bool empty(){
        return data.empty();
    }
};

int main(){
    priority_queue<char> heap;
    heap.push('5');
    heap.push('4');
    heap.push('3');
    heap.push('9');
    heap.push('6');
    while(!heap.empty()){
        cout<<heap.top()<<endl;
        heap.pop();
    }//while
}


STL里面的 priority_queue 写法与此相似,只是增加了模板及相关的迭代器什么的。 

priority_queue 对于基本类型的使用方法相对简单。他的模板声明带有三个参数:
priority_queue<Type, Container, Functional>
其中Type 为数据类型, Container 为保存数据的容器,Functional 为元素比较方式。
Container 必须是用数组实现的容器,比如 vector, deque 但不能用 list.
STL里面默认用的是 vector. 比较方式默认用 operator< , 所以如果你把后面俩个参数缺省的话,
优先队列就是大顶堆,队头元素最大。

#include <iostream>
#include <queue>
using namespace std;

int main(){
    priority_queue<char> heap;
    heap.push('5');
    heap.push('4');
    heap.push('3');
    heap.push('9');
    heap.push('6');
    // 输出最大优先级队列
    while(!heap.empty()){
        cout<<heap.top()<<endl;
        heap.pop();
    }//while
}

如果要用到小顶堆,则一般要把模板的三个参数都带进去。
STL里面定义了一个仿函数 greater<>,对于基本类型可以用这个仿函数声明小顶堆

#include <iostream>
#include <queue>
using namespace std;

int main(){
    // 最小优先级队列
    priority_queue<char,vector<char>,greater<char> > heap;
    heap.push('5');
    heap.push('4');
    heap.push('3');
    heap.push('9');
    heap.push('6');
    // 输出最大优先级队列
    while(!heap.empty()){
        cout<<heap.top()<<endl;
        heap.pop();
    }//while
}

对于自定义类型,则必须自己重载 operator< 或者自己写仿函数

【自定义优先级】

重载 operator<

[cpp]  view plain copy
  1. #include<iostream>  
  2. #include<stdio.h>  
  3. #include<queue>  
  4. using namespace std;  
  5.   
  6. struct Node    
  7. {    
  8.     //值  
  9.     int value;  
  10.     //编号  
  11.     int key;    
  12.     //重载操作符  
  13.     friend bool operator < (Node node1,Node node2)    
  14.     {      
  15.         //最大优先队列  
  16.         return node1.value < node2.value;    
  17.     }  
  18.     /* 
  19.     不要重载这个'>'只重载'<' 
  20.     friend bool operator > (Node node1,Node node2)   
  21.     {   
  22.         return node1.value > node2.value;   
  23.     }  
  24.     */  
  25. };  
  26.   
  27. struct Node2    
  28. {    
  29.     //值  
  30.     int value;  
  31.     //编号  
  32.     int key;    
  33.     //重载操作符  
  34.     friend bool operator < (Node2 node1,Node2 node2)    
  35.     {      
  36.         //最小优先队列  
  37.         return node1.value > node2.value;    
  38.     }  
  39. };  
  40.   
  41. int main(){  
  42.     int i;  
  43.     //实例一 结构体1  
  44.     Node b[5];  
  45.     b[0].value = 6; b[0].key = 1;   
  46.     b[1].value = 9; b[1].key = 2;   
  47.     b[2].value = 2; b[2].key = 3;   
  48.     b[3].value = 8; b[3].key = 4;   
  49.     b[4].value = 1; b[4].key = 5;   
  50.     //最大优先队列  
  51.     priority_queue<Node> Heap;  
  52.     //入队列  
  53.     for(i = 0;i < 5;i++){  
  54.         Heap.push(b[i]);  
  55.     }  
  56.     printf("最大优先队列:\n");  
  57.     //出队列  
  58.     for(i = 0;i < 5;i++){  
  59.         printf("key:%d value:%d\n",Heap.top().key,Heap.top().value);  
  60.         Heap.pop();  
  61.     }  
  62.     //实例二 结构体2  
  63.     Node2 b2[5];  
  64.     b2[0].value = 6; b2[0].key = 1;   
  65.     b2[1].value = 9; b2[1].key = 2;   
  66.     b2[2].value = 2; b2[2].key = 3;   
  67.     b2[3].value = 8; b2[3].key = 4;   
  68.     b2[4].value = 1; b2[4].key = 5;   
  69.     //最大优先队列  
  70.     priority_queue<Node2> Heap2;  
  71.     //入队列  
  72.     for(i = 0;i < 5;i++){  
  73.         Heap2.push(b2[i]);  
  74.     }  
  75.     printf("最小优先队列:\n");  
  76.     //出队列  
  77.     for(i = 0;i < 5;i++){  
  78.         printf("key:%d value:%d\n",Heap2.top().key,Heap2.top().value);  
  79.         Heap2.pop();  
  80.     }  
  81.     return 0;  
  82. }  



注意:

[cpp]  view plain copy
  1. struct Node    
  2. {    
  3.     //值  
  4.     int value;  
  5.     //编号  
  6.     int key;    
  7.     friend bool operator > (Node node1,Node node2)    
  8.     {    
  9.         return node1.value > node2.value;    
  10.     }   
  11. };  

这样会报错。因为标准库默认使用元素类型的<操作符来确定它们之间的优先级关系。


自定义类型重载 operator< 后,声明对象时就可以只带一个模板参数。
但此时不能像基本类型这样声明priority_queue<Node, vector<Node>, greater<Node> >;
原因是 greater<Node> 没有定义,如果想用这种方法定义则可以按如下方式:

#include <iostream>
#include <queue>
using namespace std;

struct Node{
    int value;
    int key;
    Node(int x,int y):key(x),value(y){}
};

struct cmp{
    bool operator()(Node a,Node b){
        if(a.key == b.key){
            return a.value > b.value;
        }
        return a.key > b.key;
    }
};

int main(){

    priority_queue<Node,vector<Node>,cmp> heap;
    Node node0(5,6);
    Node node1(3,3);
    Node node2(2,4);
    Node node3(2,3);
    Node node4(1,3);

    heap.push(node0);
    heap.push(node1);
    heap.push(node2);
    heap.push(node3);
    heap.push(node4);

    while(!heap.empty()){
        Node node = heap.top();
        cout<<"Key->"<<node.key<<" Value->"<<node.value<<endl;
        heap.pop();
    }//while
}




具体实例:点击打开链接

                   看病要排队

                   搬水果

目录
相关文章
|
算法
浅谈算法和数据结构: 五 优先级队列与堆排序
原文:浅谈算法和数据结构: 五 优先级队列与堆排序 在很多应用中,我们通常需要按照优先级情况对待处理对象进行处理,比如首先处理优先级最高的对象,然后处理次高的对象。最简单的一个例子就是,在手机上玩游戏的时候,如果有来电,那么系统应该优先处理打进来的电话。
1334 0
|
1月前
|
算法
【MATLAB】语音信号识别与处理:滑动平均滤波算法去噪及谱相减算法呈现频谱
【MATLAB】语音信号识别与处理:滑动平均滤波算法去噪及谱相减算法呈现频谱
45 0
|
1月前
|
传感器 算法 计算机视觉
基于肤色模型和中值滤波的手部检测算法FPGA实现,包括tb测试文件和MATLAB辅助验证
该内容是关于一个基于肤色模型和中值滤波的手部检测算法的描述,包括算法的运行效果图和所使用的软件版本(matlab2022a, vivado2019.2)。算法分为肤色分割和中值滤波两步,其中肤色模型在YCbCr色彩空间定义,中值滤波用于去除噪声。提供了一段核心程序代码,用于处理图像数据并在FPGA上实现。最终,检测结果输出到&quot;hand.txt&quot;文件。
|
1月前
|
机器学习/深度学习 算法 计算机视觉
基于yolov2深度学习网络的视频手部检测算法matlab仿真
基于yolov2深度学习网络的视频手部检测算法matlab仿真
|
1月前
|
算法
【MATLAB】语音信号识别与处理:移动中位数滤波算法去噪及谱相减算法呈现频谱
【MATLAB】语音信号识别与处理:移动中位数滤波算法去噪及谱相减算法呈现频谱
23 2
|
1月前
|
算法
【MATLAB】语音信号识别与处理:卷积滑动平均滤波算法去噪及谱相减算法呈现频谱
【MATLAB】语音信号识别与处理:卷积滑动平均滤波算法去噪及谱相减算法呈现频谱
33 0
|
1月前
|
算法
【MATLAB】语音信号识别与处理:一维信号NLM非局部均值滤波算法去噪及谱相减算法呈现频谱
【MATLAB】语音信号识别与处理:一维信号NLM非局部均值滤波算法去噪及谱相减算法呈现频谱
40 1
|
5天前
|
机器学习/深度学习 人工智能 算法
基于DCT和扩频的音频水印嵌入提取算法matlab仿真
本文介绍了结合DCT和扩频技术的音频水印算法,用于在不降低音质的情况下嵌入版权信息。在matlab2022a中实现,算法利用DCT进行频域处理,通过扩频增强水印的隐蔽性和抗攻击性。核心程序展示了水印的嵌入与提取过程,包括DCT变换、水印扩频及反变换步骤。该方法有效且专业,未来研究将侧重于提高实用性和安全性。
|
9天前
|
文字识别 算法 计算机视觉
图像倾斜校正算法的MATLAB实现:图像倾斜角检测及校正
图像倾斜校正算法的MATLAB实现:图像倾斜角检测及校正
15 0
|
12天前
|
机器学习/深度学习 算法
【MATLAB】GA_ELM神经网络时序预测算法
【MATLAB】GA_ELM神经网络时序预测算法
286 9