STL--priority_queue--自定义数据类型

STL中priority_queue的声明模板有3个参数priority_queue<Type,Container,Functional>。

当使用的数据类型Type为自定义数据类型时有以下3种方法。

1)写仿函数

 #include<iostream>
 #include<queue>
 using namespace std;
 struct Node
 {
     int x,y;
 };
 struct cmp
 {
     bool operator()(Node a,Node b)
     {
         if(a.x!=b.x)
             return a.x<b.x;//<为大顶堆，>为小顶堆
         return a.y<b.y;
     }
 };
 int main()
 {
     Node node[]={{,},{,},{,},{,},{,},{,},{,},{,},{,}};
     //建立9个结点
     priority_queue <Node,vector<Node>,cmp> q(node,node+);//将9个点存入优先队列q
     while(!q.empty())//输出
     {
         Node n=q.top();
         cout<<n.x<<' '<<n.y<<endl;
         q.pop();
     }
     return ;
 }

2)数据类型外重载operator<

 #include<iostream>
 #include<queue>
 using namespace std;
 struct Node
 {
     int x,y;
 };
 bool operator<(Node a,Node b)
 {
     if(a.x!=b.x)
         return a.x<b.x;//<为大顶堆，>为小顶堆
     return a.y<b.y;
 }
 int main()
 {
     Node node[]={{,},{,},{,},{,},{,},{,},{,},{,},{,}};
     //建立9个结点
     priority_queue <Node> q(node,node+);//将9个点存入优先队列q
     while(!q.empty())//输出
     {
         Node n=q.top();
         cout<<n.x<<' '<<n.y<<endl;
         q.pop();
     }
     return ;
 }

3)数据类型内重载operator<

 #include<iostream>
 #include<queue>
 using namespace std;
 struct Node
 {
     int x,y;
     bool operator<(const Node a)const
     {
         if(x!=a.x)
             return x<a.x;
         return y<a.y;
     }
 };
 int main()
 {
     Node node[]={{,},{,},{,},{,},{,},{,},{,},{,},{,}};
     //建立9个结点
     priority_queue <Node> q(node,node+);//将9个点存入优先队列q
     while(!q.empty())//输出
     {
         Node n=q.top();
         cout<<n.x<<' '<<n.y<<endl;
         q.pop();
     }
     return ;
 }