STL之priority_queue

最新推荐文章于 2024-09-19 20:56:21 发布

原创最新推荐文章于 2024-09-19 20:56:21 发布 · 282 阅读

0 ·

CC 4.0 BY-SA版权

STL 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了如何使用C++标准库中的优先队列(priority_queue)来实现大根堆和小根堆。通过三种不同的方式展示了优先队列的基本用法，并且自定义了一个Node结构体来演示如何处理复杂的数据类型。

部署运行你感兴趣的模型镜像

#include <iostream>  
#include<functional>
#include <queue>  

using namespace std;  

struct Node{  
    int x;  
    Node( int a= 0):x(a) {}  
    friend bool operator<(Node a,Node b )
    {  
       return a.x<b.x;//  使用">"则为小根堆  
    }  
};  

struct cmp
{
    bool operator()(Node a,Node b)
    {
        return a.x<b.x;
    }
};
int main(){  

    int data2[5]={1,3,7,-4,56}; 
    /******方法一*/
    priority_queue<int> big(data2,data2+5); 
     while(!big.empty())
    {  
        cout<<big.top()<<' ';  
        big.pop();  
    }  
    cout<<endl;  
    /******方法二*/
    priority_queue<int,vector<int>,cmp> little(data2,data2+5);  
    //priority_queue<int,vector<int>,greater<int>> little(data2,data2+5);  
    while(!little.empty())
    {  
        cout<<little.top()<<' ';  
        little.pop();  
    }  
    cout<<endl;  
    /******方法三*/
    Node data1[10];  
    for( int i= 0; i< 10; ++i )  
      data1[i]=Node(rand());  
    priority_queue<Node> q(data1,data1+10);//自定义大根模式      
    while( !q.empty() )
    {  
        cout << q.top().x << ' ' ;  
        q.pop();  
    } 
    cout<<endl;
    return 0;  
}