priority

xiaoxiao2021-02-28  202

推荐阅读: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; }

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>> 默认比较第一个值

转载请注明原文地址: https://www.6miu.com/read-27810.html

最新回复(0)