推荐阅读:C++ priority_queue 与 lambda的结合使用
#include <queue> 头文件
priority_queue<Type, Container, Functional>
默认是大顶堆
greater<>成为小顶堆priority_queue<
int
, vector<
int
>, greater<
int
> > q;
加入 q.push();
弹出 q.pop();
队头 q.top();
对于自定义类型,则必须自己重载 operator< 或者自己写仿函数
这里得仿函数做的是判断当前值得大小往后放,因此,greater<>,大的元素往后放,就是小顶堆。
struct Node{
int x, y;
Node(int a= 0, int b= 0 ):
x(a), y(b) {}
};
bool operator<( Node a, Node b ){
if( a.x== b.x ) return a.y> b.y;//小顶堆
return a.x> b.x;
}
#include <iostream>
#include <queue>
using namespace std;
struct Node{
int x, y;
Node( int a= 0, int b= 0 ):
x(a), y(b) {}
};
struct cmp{
bool operator() ( Node a, Node b ){
if( a.x== b.x ) return a.y> b.y;
return a.x> b.x; }// 小顶堆
};
int main(){
priority_queue<Node, vector<Node>, cmp> q;
for( int i= 0; i< 10; ++i )
q.push( Node( rand(), rand() ) );
while( !q.empty() ){
cout << q.top().x << ' ' << q.top().y << endl;
q.pop();
}
getchar();
return 0;
}
例题:
1、美团2016 字符编码
请设计一个算法,给一个字符串进行二进制编码,使得编码后字符串的长度最短。
输入描述:
每组数据一行,为待编码的字符串。保证字符串长度小于等于1000。
输出描述:
一行输出最短的编码后长度。
输入例子1:
MT-TECH-TEAM
输出例子1:
33
自己的思路:
自己独立更改了很多次,核心的方法是哈夫曼编码。首先使用一个map统计所有字符出现的次数,放入一个小顶堆,堆的值按照次数排序。每次弹出两个最小的两个,构建成一个节点后再加入堆中。当堆为空时,说明所有节点都处理完成。
更改很多次的原因是,怎么迎合题目输出的要求。举例:一个节点的次数是2,当他和其他节点构成一个新的节点时,编码长度的改变是:这个节点对应的两个相同字符都加一。
因此,堆中存储的数据更改为一个结构体{sum,node},sum表示这个节点目前的编码长度统计(默认带有前支),node代表当前节点的权值;
每次构建一个新节点时,新节点的sum=两个节点的sum之和,同时要加上两个节点的权值之和(因为有产生一个向上的分支,这个权值代表的是当前这个综合节点的字符数量),新的节点探入{新的sum,权值之和}
特殊情况:当队列为空时,只有一个节点了,这个时候的sum只是两个sum的和,不能再加上两个节点的权值和(根节点没有向上的分支)
代码如下:
#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<map>
using namespace std;
struct code{
int sum;
int node;
};
bool operator<(code a,code b){
if(a.node==b.node) return a.sum>b.sum;
return a.node>b.node;
}
int main()
{
//哈夫曼编码
string input;
while(cin>>input)
{
map<char,int> sta;
priority_queue<code,vector<code>> q;
for(auto s:input)
{
sta[s]++;
}
for(auto s:sta)
q.push({s.second,s.second});
int n=q.size()-1;
int sum=0;
if(n==0) sum=1;
while(n!=0&&!q.empty())
{
code left=q.top();
q.pop();
code right=q.top();
q.pop();
int temp=left.sum+right.sum+left.node+right.node;
if(!q.empty())
q.push({temp,left.node+right.node});
if(q.empty())
sum=temp-left.node-right.node;//到根节点,将向上分支多加的减掉
}
cout<<sum<<endl;
}
//system("pause");
return 0;
}
2 378. Kth Smallest Element in a Sorted Matrix
http://blog.csdn.net/m0_37693059/article/details/77772048
使用了
priority_queue<pair<int, pair<int, int>>> myQueue;
make_pair ,emplace,lambda表达式等
greater<pair<int,int>> 默认比较第一个值